mindspore.ops.trace

mindspore.ops.trace(input)[源代码]

返回tensor对角线的元素总和。

说明

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

参数：

input (Tensor) - 输入tensor。

返回：

Tensor

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore
>>> input = mindspore.tensor([[0, 1, 2], 
...                           [3, 4, 5], 
...                           [6, 7, 8]])
>>> mindspore.ops.trace(input)
Tensor(shape=[], dtype=Int64, value= 12)