未验证 提交 df08209a 编写于 作者: Z zhaoyuchen2018 提交者: GitHub

[cherry-pick] Add descending parameter for argsort doc. (#1621) (#1632)

* Add descending parameter for argsort doc.

* Refine doc

* Refine format
上级 ae778ef6
......@@ -3,14 +3,15 @@
argsort
-------------------------------
.. py:function:: paddle.fluid.layers.argsort(input,axis=-1,name=None)
.. py:function:: paddle.fluid.layers.argsort(input,axis=-1,descending=False,name=None)
对输入变量沿给定轴进行 **升序** 排列,输出排序好的数据和相应的索引,其维度和输入相同。**暂不支持降序排列**
对输入变量沿给定轴进行排序,输出排序好的数据和相应的索引,其维度和输入相同。**默认升序排列,如果需要降序排列设置** ``descending=True``
参数:
- **input** (Variable) - 输入的多维 ``Tensor`` ,支持的数据类型:float32、float64。
- **axis** (int,可选) - 指定对输入Tensor进行运算的轴, ``axis`` 的有效范围是[-R, R),R是输入 ``x`` 的Rank, ``axis`` 为负时与 ``axis`` +R 等价。默认值为0。
- **descending** (bool,可选) - 指定算法排序的方向。如果设置为True,算法按照降序排序。如果设置为False或者不设置,按照升序排序。默认值为False。
- **name** (str,可选) – 具体用法请参见 :ref:`api_guide_Name` ,一般无需设置,默认值为None。
返回:一组已排序的输出(与 ``input`` 维度相同、数据类型相同)和索引(数据类型为int64)。
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册