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.
is the main diagonal, while is below it, and 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
for , 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.]]