forked from mindspore-Ecosystem/mindspore
19 lines
843 B
ReStructuredText
19 lines
843 B
ReStructuredText
mindspore.ops.diagonal
|
||
=======================
|
||
|
||
.. py:function:: mindspore.ops.diagonal(input, offset=0, dim1=0, dim2=1)
|
||
|
||
返回 `input` 的部分视图，其相对于 `dim1` 和 `dim2` 的对角线元素作为维度附加在形状的末尾。
|
||
|
||
参数：
|
||
- **input** (Tensor) - 输入Tensor，其维度至少为2。
|
||
- **offset** (int, 可选) - 对角线与主对角线的偏移。可以是正值或负值。默认值：0。
|
||
- **dim1** (int, 可选) - 二维子数组的第一轴，对角线应该从这里开始。默认值：0。
|
||
- **dim2** (int, 可选) - 二维子数组的第二轴，对角线应该从这里开始。默认值：1。
|
||
|
||
返回：
|
||
Tensor，如果Tensor是二维，则返回值是一维数组。
|
||
|
||
异常：
|
||
- **ValueError** - 输入Tensor的维度少于2。
|