mindspore.ops.Trace

class mindspore.ops.Trace

返回在Tensor的对角线方向上的总和。

说明

输入必须是Tensor，复数类型暂不支持。

警告

这是一个实验性API，后续可能修改或删除。

输入：

• x (Tensor) - 二维Tensor。

输出：

Tensor，含有一个元素的零维Tensor，其数据类型与 x 一致。

异常：

• TypeError - 如果 x 不是Tensor。

• ValueError - 如果当 x 的维度不是2。

支持平台：

Ascend GPU CPU

样例：

>>> x = Tensor(np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]]), mindspore.float32)
>>> trace = ops.Trace()
>>> output = trace(x)
>>> print(output)
15.0
>>> x = Tensor(np.arange(1, 13).reshape(3, 4), mindspore.float32)
>>> trace = ops.Trace()
>>> output = trace(x)
>>> print(output)
18.0
>>> x = Tensor(np.arange(12, 0, -1).reshape(4, 3), mindspore.float32)
>>> trace = ops.Trace()
>>> output = trace(x)
>>> print(output)
24.0