cupy.triu_indices#

cupy.triu_indices(n, k=0, m=None)[source]#

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

参数：

n (int) – 返回索引有效的数组大小。
k (int, optional) – 指对角线偏移。默认情况下，k = 0，即主对角线。k 的正值表示主对角线上方的对角线，负值表示主对角线下方的对角线。
m (int, optional) – 返回数组有效的数组的列维度。默认情况下，m = n。

返回：

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

返回类型：

tuple of ndarrays

另请参阅