mindspore.mint.trace

mindspore.mint.trace(input)

返回 input 的主对角线方向上的总和。

参数：

• input (Tensor) - 二维输入tensor。

返回：

Tensor，当 input 为数据类型为整型或bool时其数据类型为mindspore.int64，反之与 input 一致，含有一个元素。

异常：

• ValueError - input 的维度不是2。

支持平台：

Ascend

样例：

>>> import mindspore
>>> input = mindspore.tensor(mindspore.mint.arange(1, 13).reshape(3, 4), mindspore.float32)
>>> print(input)
[[ 1.  2.  3.  4.]
 [ 5.  6.  7.  8.]
 [ 9. 10. 11. 12.]]
>>> output = mindspore.mint.trace(input)
>>> print(output)
18.0