cupy.diagonal

cupy.diagonal(a, offset=0, axis1=0, axis2=1)

返回指定的对角线。

此函数提取沿两个指定轴的对角线。其他轴不变。此函数返回此数组的可写视图，如同 NumPy 1.10 所做的那样。

参数:

• a (cupy.ndarray) – 从中提取对角线的数组。

• offset (int) – 对角线索引。零表示主对角线，正值表示上对角线，负值表示下对角线。

• axis1 (int) – 第一个从中取对角线的轴。

• axis2 (int) – 第二个从中取对角线的轴。

返回:

a 的对角线视图。

返回类型:

cupy.ndarray

参见

numpy.diagonal()