cupy.tril_indices

cupy.tril_indices(n, k=0, m=None)

返回下三角形矩阵的索引。其中，第一组元素包含所有索引的行坐标，第二组元素包含列坐标。

参数:

• n (int) – 返回的索引有效的数组的行维度。

• k (int, 可选) – 零元素的对角线之上。 k = 0 (默认) 是主对角线， k < 0 在主对角线下方， k > 0 在主对角线上方。

• m (int, 可选) – 返回的数组有效的数组的列维度。默认情况下，m = n。

返回:

y – 三角形的索引。返回的元组包含两个数组，每个数组包含数组沿一个维度的索引。

返回类型:

tuple (包含 ndarrays)

另请参阅

numpy.tril_indices