diff --git a/docs/api/paddle/argmin_cn.rst b/docs/api/paddle/argmin_cn.rst index 23aac41f7e8..aafd2951fdc 100644 --- a/docs/api/paddle/argmin_cn.rst +++ b/docs/api/paddle/argmin_cn.rst @@ -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 类型的结果。