Skip to content
Open
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 6 additions & 1 deletion docs/api/paddle/diag_cn.rst
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
diag
-------------------------------

.. py:function:: paddle.diag(x, offset=0, padding_value=0, name=None)
.. py:function:: paddle.diag(x, offset=0, padding_value=0, name=None, *, out=None)


如果 ``x`` 是向量(1-D Tensor),则返回带有 ``x`` 元素作为对角线的 2-D 方阵。
Expand All @@ -16,12 +16,17 @@ diag
- 如果 ``offset`` > 0,则为上对角线。
- 如果 ``offset`` < 0,则为下对角线。

.. note::

别名支持: 参数名 ``input`` 可替代 ``x``, ``diagonal`` 可替代 ``offset``

参数
:::::::::
- **x** (Tensor) - 输入的 `Tensor`。它的形状可以是一维或二维。其数据类型应为 float16、float32、float64、int32、int64、complex64、complex128。
- **offset** (int,可选) - 对角线偏移量。正值表示上对角线,0 表示主对角线,负值表示下对角线。默认值为 0。
- **padding_value** (int|float,可选) -使用此值来填充指定对角线以外的区域。仅在输入为一维 Tensor 时生效。默认值为 0。
- **name** (str,可选) - 具体用法请参见 :ref:`api_guide_Name`,一般无需设置,默认值为 None。
- **out** (Tensor,可选)- 输出的结果。该参数为仅关键字参数,默认值为 None。

返回
:::::::::
Expand Down