• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1mindspore.ops.FillDiagonal
2==========================
3
4.. py:class:: mindspore.ops.FillDiagonal(fill_value, wrap=False)
5
6    填充至少具有二维的Tensor的主对角线。当输入的维度大于2时，所有维度必须等长。此函数原地修改输入Tensor，并返回输入Tensor。
7
8    .. warning::
9        这是一个实验性API，后续可能修改或删除。
10
11    参数：
12        - **fill_value** (float) - `input_x` 对角线的填充值。
13        - **wrap** (bool，可选) - 控制对于一个高矩阵（即矩阵的行数大于列数），对角线元素是否继续延伸到剩余的行。具体效果详见下方代码样例。默认值： ``False`` 。
14
15    输入：
16        - **input_x** (Tensor) - shape为 :math:`(x_1, x_2, ..., x_R)` 。
17
18    输出：
19        - **y** (Tensor) - Tensor，和输入 `input_x` 具有相同的shape和dtype。
20
21    异常：
22        - **ValueError** - 如果 `input_x` 的维度没有大于1。
23        - **ValueError** - 当维度大于2时，每个轴的大小不相等。
24