cupy.diag_indices_from

cupy.diag_indices_from(arr)

返回用于访问 N 维数组主对角线的索引。详细信息请参阅 diag_indices。

参数:

arr (cupy.ndarray) – 至少为 2 维。

另请参阅

numpy.diag_indices_from()