mindspore.numpy.tril

查看源文件
mindspore.numpy.tril(m, k=0)[源代码]

返回张量的下三角部分。 返回Tensor的副本,其中第 k 个对角线以上的元素为0。

参数:
  • m (Union[Tensor, list, tuple]) - 输入的Tensor。返回的Tensor与 m 的shape、数据类型都相同。

  • k (int, 可选) - 对角线的偏移量: \(k=0\) 即为主对角线, \(k<0\) 即对角线向下偏移, \(k>0\) 即对角线向上偏移。默认值: 0

返回:

Tensor, m 的下三角部分,shape和数据类型与 m 相同。

异常:
  • TypeError - 如果输入参数非上述给定的类型。

  • ValueError - 如果输入的 m 的秩 \(<1\)

支持平台:

Ascend GPU CPU

样例:

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