返回指定的对角线元素。
如果 a 是二维数组，则返回一个包含给定 offset 处对角线元素的一维数组。
如果 a 的维度大于二维，则 axis1 和 axis2 确定提取对角线的二维子数组。新的形状是原始形状移除 axis1 和 axis2 后，并在末尾插入一个对应于对角线的新维度。
- 参数:
a (array) – 输入数组
offset (int, 可选) – 对角线相对于主对角线的偏移量。可以是正数或负数。默认值: 0。
axis1 (int, 可选) – 提取对角线的二维子数组的第一个轴。默认值: 0。
axis2 (int, 可选) – 提取对角线的二维子数组的第二个轴。默认值: 1。
- 返回:
数组的对角线元素。
- 返回类型:
array
