Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[fp16] suppot fp16 in std,var,tensordot,diagflat #5668

Merged
merged 1 commit into from
Mar 2, 2023
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion docs/api/paddle/diagflat_cn.rst
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@ diagflat

参数
:::::::::
- **x** (Tensor) - 输入的 `Tensor`。它的形状可以是任意维度。其数据类型应为 float32,float64,int32,int64。
- **x** (Tensor) - 输入的 `Tensor`。它的形状可以是任意维度。其数据类型应为 float16,float32,float64,int32,int64。
- **offset** (int,可选) - 对角线偏移量。正值表示上对角线,0 表示主对角线,负值表示下对角线。
- **name** (str,可选) - 具体用法请参见 :ref:`api_guide_Name`,一般无需设置,默认值为 None。

Expand Down
2 changes: 1 addition & 1 deletion docs/api/paddle/std_cn.rst
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@ std

参数
::::::::::
- **x** (Tensor) - 输入的 Tensor,数据类型为:float32、float64。
- **x** (Tensor) - 输入的 Tensor,数据类型为:float16、float32、float64。
- **axis** (int|list|tuple,可选) - 指定对 ``x`` 进行计算的轴。``axis`` 可以是 int、list(int)、tuple(int)。如果 ``axis`` 包含多个维度,则沿着 ``axis`` 中的所有轴进行计算。``axis`` 或者其中的元素值应该在范围[-D, D)内,D 是 ``x`` 的维度。如果 ``axis`` 或者其中的元素值小于 0,则等价于 :math:`axis + D`。如果 ``axis`` 是 None,则对 ``x`` 的全部元素计算标准差。默认值为 None。
- **unbiased** (bool,可选) - 是否使用无偏估计来计算标准差。使用 :math:`N` 来代表在 axis 上的维度,如果 ``unbiased`` 为 True,则在计算中使用 :math:`N - 1` 作为除数。为 False 时将使用 :math:`N` 作为除数。默认值为 True。
- **keepdim** (bool,可选) - 是否在输出 Tensor 中保留减小的维度。如果 ``keepdim`` 为 True,则输出 Tensor 和 ``x`` 具有相同的维度(减少的维度除外,减少的维度的大小为 1)。否则,输出 Tensor 的形状会在 ``axis`` 上进行 squeeze 操作。默认值为 False。
Expand Down
2 changes: 1 addition & 1 deletion docs/api/paddle/tensordot_cn.rst
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@ Tensor 缩并运算(Tensor Contraction),即沿着 axes 给定的多个轴
参数
::::::::::::

- **x** (Tensor)- 缩并运算操作的左 Tensor,数据类型为 ``float32`` 或 ``float64``。
- **x** (Tensor)- 缩并运算操作的左 Tensor,数据类型为 ``float16`` 或 ``float32`` 或 ``float64``。
- **y** (Tensor)- 缩并运算操作的右 Tensor,与 ``x`` 具有相同的数据类型。
- **axes** (int|tuple|list|Tensor)- 指定对 ``x`` 和 ``y`` 做缩并运算的轴,默认值为整数 2。

Expand Down
2 changes: 1 addition & 1 deletion docs/api/paddle/var_cn.rst
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@ var

参数
::::::::::
- **x** (Tensor) - 输入的 Tensor,数据类型为:float32、float64。
- **x** (Tensor) - 输入的 Tensor,数据类型为:float16、float32、float64。
- **axis** (int|list|tuple,可选) - 指定对 ``x`` 进行计算的轴。``axis`` 可以是 int、list(int)、tuple(int)。

- 如果 ``axis`` 包含多个维度,则沿着 ``axis`` 中的所有轴进行计算。``axis`` 或者其中的元素值应该在范围[-D, D)内,D 是 ``x`` 的维度。
Expand Down