mindspore.numpy.triu_indices

mindspore.numpy.triu_indices(n, k=0, m=None)

Returns the indices for the upper-triangle of an (n, m) array.

参数

• n (int) – The size of the arrays for which the returned indices will be valid.

• k (int, optional) – Diagonal offset, default: 0 .

• m (int, optional) – The column dimension of the arrays for which the returned arrays will be valid. By default m is taken equal to n.

返回

The indices for the triangle. The returned tuple contains two tensors, each with the indices along one dimension of the tensor.

异常

TypeError – If n, k, m are not numbers.

Supported Platforms:

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]))