cupy.diagonal#

cupy.diagonal(a, offset=0, axis1=0, axis2=1)[source]#

返回指定的对角线元素。

此函数提取沿两个指定轴的对角线元素。其他轴不变。此函数返回数组的一个可写视图，与 NumPy 1.10 的行为一致。

参数:

a (cupy.ndarray) – 用于提取对角线元素的数组。
offset (int) – 对角线的索引。零表示主对角线，正值表示上对角线，负值表示下对角线。
axis1 (int) – 用于提取对角线的第一个轴。
axis2 (int) – 用于提取对角线的第二个轴。

返回值:

a 的对角线视图。

返回类型:

另请参阅

numpy.diagonal()