# mindspore.numpy.tri¶

mindspore.numpy.tri(N, M=None, k=0, dtype=mstype.float32)[source]

Returns a tensor with ones at and below the given diagonal and zeros elsewhere.

Parameters
• N (int) – Number of rows in the array.

• M (int, optional) – Number of columns in the array. By default, M is taken equal to N.

• k (int, optional) – The sub-diagonal at and below which the array is filled. $$k = 0$$ is the main diagonal, while $$k < 0$$ is below it, and $$k > 0$$ is above. The default is 0.

• dtype (mindspore.dtype, optional) – Data type of the returned array. The default is mstype.float32.

Returns

Tensor with shape (N, M), with its lower triangle filled with ones and zeros elsewhere; in other words $$T[i,j] = 1$$ for $$j <= i + k$$, 0 otherwise.

Raises

TypeError – If input arguments have types not specified above.

Supported Platforms:

Ascend GPU CPU

Examples

>>> import mindspore.numpy as np
>>> output = np.tri(3, 3, 1)
>>> print(output)
[[1. 1. 0.]
[1. 1. 1.]
[1. 1. 1.]]