mindspore.ops.Triu

class mindspore.ops.Triu(diagonal=0)

返回单个或一批二维矩阵上三角形部分，其他位置的元素将被置零。 矩阵的上三角形部分定义为对角线和对角线上方的元素。

警告

这是一个实验性API，后续可能修改或删除。

参数：

• diagonal (int，可选) - 可选参数指定对角线位置，默认值： 0 ，指定主对角线。

输入：

• x (Tensor) - 输入Tensor。shape为 \((M, N, *)\) ，其中 \(*\) 为任意数量的额外维度。其数据类型为数值型。

输出：

• y (Tensor) - 数据类型和shape维度与输入相同。

异常：

• TypeError - 如果 x 不是Tensor。

• TypeError - 如果 diagonal 不是int类型。

• ValueError - 如果 x 的shape长度小于2。

支持平台：

Ascend GPU CPU

样例：

>>> x = Tensor(np.array([[ 1,  2,  3,  4],
...                      [ 5,  6,  7,  8],
...                      [10, 11, 12, 13],
...                      [14, 15, 16, 17]]))
>>> triu = ops.Triu()
>>> result = triu(x)
>>> print(result)
[[ 1  2  3  4]
 [ 0  6  7  8]
 [ 0  0 12 13]
 [ 0  0  0 17]]
>>> x = Tensor(np.array([[ 1,  2,  3,  4],
...                      [ 5,  6,  7,  8],
...                      [10, 11, 12, 13],
...                      [14, 15, 16, 17]]))
>>> triu = ops.Triu(diagonal=1)
>>> result = triu(x)
>>> print(result)
[[ 0  2  3  4]
 [ 0  0  7  8]
 [ 0  0  0 13]
 [ 0  0  0  0]]
>>> x = Tensor(np.array([[ 1,  2,  3,  4],
...                      [ 5,  6,  7,  8],
...                      [10, 11, 12, 13],
...                      [14, 15, 16, 17]]))
>>> triu = ops.Triu(diagonal=-1)
>>> result = triu(x)
>>> print(result)
[[ 1  2  3  4]
 [ 5  6  7  8]
 [ 0 11 12 13]
 [ 0  0 16 17]]