diff --git a/doc/fluid/api_cn/tensor_cn/argmax_cn.rst b/doc/fluid/api_cn/tensor_cn/argmax_cn.rst index 157806c38000bd4ae1866e55e4ecbe5bd90e7882..d2fb738b4cd99c1cf6d458ed994aae2b7186f09d 100644 --- a/doc/fluid/api_cn/tensor_cn/argmax_cn.rst +++ b/doc/fluid/api_cn/tensor_cn/argmax_cn.rst @@ -3,7 +3,7 @@ argmax ------------------------------- -.. py:function:: paddle.argmax(x, axis=None, dtype=None, keepdim=False, name=None) +.. py:function:: paddle.argmax(x, axis=None, keepdim=False, dtype='int64', name=None) 该OP沿 ``axis`` 计算输入 ``x`` 的最大元素的索引。 @@ -12,8 +12,8 @@ argmax :::::::: - **x** (Tensor) - 输入的多维 ``Tensor`` ,支持的数据类型:float32、float64、int16、int32、int64、uint8。 - **axis** (int,可选) - 指定对输入Tensor进行运算的轴, ``axis`` 的有效范围是[-R, R),R是输入 ``x`` 的维度个数, ``axis`` 为负数时,进行计算的 ``axis`` 与 ``axis`` + R 一致。默认值为None, 将会对输入的 `x` 进行平铺展开,返回最大值的索引。 - - **dtype** (np.dtype|str)- 输出Tensor的数据类型,可选值为int32,int64,默认值为None,将返回int64类型的结果。 - **keepdim** (bool,可选)- 是否保留进行最大值索引操作的轴,默认值为False。 + - **dtype** (np.dtype|str,可选)- 输出Tensor的数据类型,可选值为int32,int64,默认值为int64,将返回int64类型的结果。 - **name** (str,可选) – 具体用法请参见 :ref:`api_guide_Name` ,一般无需设置,默认值为None。 返回 diff --git a/doc/fluid/api_cn/tensor_cn/argmin_cn.rst b/doc/fluid/api_cn/tensor_cn/argmin_cn.rst index 14082c452382ba2e7c17eb11306e190b087e0c70..4fd851764cbffe8a5ba93e3605034a9c55854e47 100644 --- a/doc/fluid/api_cn/tensor_cn/argmin_cn.rst +++ b/doc/fluid/api_cn/tensor_cn/argmin_cn.rst @@ -3,7 +3,7 @@ argmin ------------------------------- -.. py:function:: paddle.argmin(x, axis=None, dtype=None, keepdim=False, name=None) +.. py:function:: paddle.argmin(x, axis=None, keepdim=False, dtype='int64', name=None) 该OP沿 ``axis`` 计算输入 ``x`` 的最小元素的索引。 @@ -12,8 +12,8 @@ argmin :::::::: - **x** (Tensor) - 输入的多维 ``Tensor`` ,支持的数据类型:float32、float64、int16、int32、int64、uint8。 - **axis** (int,可选) - 指定对输入Tensor进行运算的轴, ``axis`` 的有效范围是[-R, R),R是输入 ``x`` 的维度个数, ``axis`` 为负数时,进行计算的 ``axis`` 与 ``axis`` + R 一致。默认值为None, 将会对输入的 `x` 进行平铺展开,返回最小值的索引。 - - **dtype** (np.dtype|str)- 输出Tensor的数据类型,可选值为int32,int64,默认值为None,将返回int64类型的结果。 - **keepdim** (bool,可选)- 是否保留进行最小值索引操作的轴,默认值为False。 + - **dtype** (np.dtype|str, 可选)- 输出Tensor的数据类型,可选值为int32,int64,默认值为'int64',将返回int64类型的结果。 - **name** (str,可选) – 具体用法请参见 :ref:`api_guide_Name` ,一般无需设置,默认值为None。 返回