未验证 提交 effa1c93 编写于 作者: W wawltor 提交者: GitHub

fix the argmax, argmin doc

fix the argmax, argmin doc, change the default value for the argmax/argmin
上级 9b9f9be6
......@@ -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。
返回
......
......@@ -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。
返回
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册