diagonal¶

paddle. diagonal ( x, offset=0, axis1=0, axis2=1, name=None ) ¶

根据参数 offset、axis1、axis2，返回输入 Tensor 的局部视图。

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

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

参数 offset 确定从指定的二维平面中获取对角线的位置：

• 如果 offset = 0，则取主对角线。

• 如果 offset > 0，则取主对角线右上的对角线。

• 如果 offset < 0，则取主对角线左下的对角线。