mindspore.numpy.tri
- mindspore.numpy.tri(N, M=None, k=0, dtype=mstype.float32)[源代码]
返回一个Tensor,在给定的对角线处及以下元素值为1,在其他位置为0。
- 参数:
N (int) - 输入数组的行数。
M (int, 可选) - 输入数组的列数。默认情况下
M
等于N
。k (int, 可选) - 对角线的偏移量:
即为主对角线, 即对角线向下偏移, 即对角线向上偏移。默认值:0
。dtype (mindspore.dtype, 可选) - 指定的Tensor数据类型。默认值:
mstype.float32
。
- 返回:
Tensor,shape为
(N, M)
,其中他的下三角区域填充为1,其余位置填充为0;用公式表达就是当 时, ,否则为0。- 异常:
TypeError - 如果输入参数非上述给定的类型。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> import mindspore.numpy as np >>> output = np.tri(3, 3, 1) >>> print(output) [[1. 1. 0.] [1. 1. 1.] [1. 1. 1.]]