trace

该 OP 计算输入 Tensor 在指定平面上的对角线元素之和，并输出相应的计算结果。

如果输入是 2D Tensor，则返回对角线元素之和。

如果输入的维度大于 2D，则返回一个由对角线元素之和组成的数组，其中对角线从由 axis1 和 axis2 指定的二维平面中获得。默认由输入的前两维组成获得对角线的 2D 平面。

• axis1 （int ， 可选）- 获取对角线的二维平面的第一维，默认值为 0。

• axis2 （int ， 可选）- 获取对角线的二维平面的第二维，默认值为 1。

Tensor,指定二维平面的对角线元素之和。数据类型和输入数据类型一致。