mindspore.numpy.triu

mindspore.numpy.triu(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.triu(np.ones((3, 3)))
>>> print(output)
[[1. 1. 1.]
[0. 1. 1.]
[0. 0. 1.]]