mindspore.ops.trace
- mindspore.ops.trace(input)[源代码]
返回 input 的主对角线方向上的总和。
说明
输入必须是Tensor,复数类型暂不支持。
- 参数:
input (Tensor) - 二维Tensor。
- 返回:
Tensor,其数据类型与 input 一致,含有一个元素。
- 异常:
TypeError - 如果 input 不是Tensor。
ValueError - 如果当 input 的维度不是2。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> import mindspore >>> import numpy as np >>> from mindspore import Tensor, ops >>> input = Tensor(np.array([[10, 11, 12], [13, 14, 15], [16, 17, 18]]), mindspore.float32) >>> output = ops.trace(input) >>> print(output) 42.0 >>> input = Tensor(np.arange(1, 13).reshape(3, 4), mindspore.float32) >>> output = ops.trace(input) >>> print(output) 18.0 >>> input = Tensor(np.arange(12, 0, -1).reshape(4, 3), mindspore.float32) >>> output = ops.trace(input) >>> print(output) 24.0