mindspore.numpy.triu_indices

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

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

Parameters

n (int) – The size of the arrays for which the returned indices will be valid.
k (int, optional) – Diagonal offset.
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.

Returns

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

Raises

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

Supported Platforms:: Ascend GPU CPU

Examples

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