mindspore.numpy.triu_indices

mindspore.numpy.triu_indices(n, k=0, m=None)[源代码]

返回shape为 (n, m) 数组的上三角的索引。

参数：

n (int) - 指定返回索引适用的数组大小。
k (int, 可选) - 对角线的偏移量。默认值： 0 。
m (int, 可选) - 指定返回索引适用的数组的列维度，默认情况下 m 等于 n 。

返回：

上三角形的索引。返回的Tuple包含两个Tensor，每个Tensor分别对应其在一个维度上的索引。

异常：

TypeError - 如果 n ， k ， m 不是数字。

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore.numpy as np
>>> print(np.triu_indices(3))
(Tensor(shape=[6], dtype=Int32, value= [0, 0, 0, 1, 1, 2]),
 Tensor(shape=[6], dtype=Int32, value= [0, 1, 2, 1, 2, 2]))