Skip to content

Commit

Permalink
Update argmin_cn.rst (#5629)
Browse files Browse the repository at this point in the history
add x support float16
  • Loading branch information
enkilee authored Feb 27, 2023
1 parent 8a45a1d commit 9aff380
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docs/api/paddle/argmin_cn.rst
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@ argmin

参数
::::::::
- **x** (Tensor) - 输入的多维 ``Tensor``,支持的数据类型:float32、float64、int16、int32、int64、uint8。
- **x** (Tensor) - 输入的多维 ``Tensor``,支持的数据类型:float16、float32、float64、int16、int32、int64、uint8。
- **axis** (int,可选) - 指定对输入 Tensor 进行运算的轴, ``axis`` 的有效范围是[-R, R),R 是输入 ``x`` 的维度个数, ``axis`` 为负数时,进行计算的 ``axis`` 与 ``axis`` + R 一致。默认值为 None,将会对输入的 `x` 进行平铺展开,返回最小值的索引。
- **keepdim** (bool,可选) - 是否保留进行最小值索引操作的轴,默认值为 False。
- **dtype** (np.dtype|str,可选) - 输出 Tensor 的数据类型,可选值为 int32、int64,默认值为'int64',将返回 int64 类型的结果。
Expand Down

0 comments on commit 9aff380

Please sign in to comment.