mindspore.ops.triu

mindspore.ops.triu(input, diagonal=0)[源代码]

将指定对角线下方的元素设置为0。

参数:
  • input (Tensor) - 输入tensor。

  • diagonal (int,可选) - 二维tensor的指定对角线。默认 0 ,表示主对角线。

返回:

Tensor

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore
>>> input = mindspore.tensor([[ vpn梯子 1,  2,  3,  4],
...        vpn永久免费梯子   vpn梯子 免费                  [ 5,  6,  7,  8],
...                           [10, vpn free 11, vpn梯子 免费 12, 13],
...     免费的vpn梯子   vpn梯子 免费                     [14, 15, 16, 17]])
>>> mindspore.ops.triu(input)
Tensor(shape=[4, 4], dtype=Int64, value=
[[ 1,  2,  3,  4],
 [ 免费的vpn梯子 0,  免费的vpn梯子 6,  7,  8],
 [ vpn梯子 0,  0, 12, 13],
 [ 0,  0,  0, 17]])
>>> mindspore.ops.triu(input, 1)
Tensor(shape=[4, 4], dtype=Int64, value=
[[ 0,  2,  3,  4],
 [ 0,  0,  7,  8],
 [ 0,  0,  0, vpn永久免费梯子 13],
 [ 0,  0,  0,  vpn梯子 免费 0]])
>>> mindspore.ops.triu(input, -1)
Tensor(shape=[4, 4], dtype=Int64, value=
[[ 1,  2,  3,  4],
 [ 5,  6, vpn永久免费梯子  7,  8],
 [ 0, 11, 12, 13],
 [ 0,  0, 16, 17]])
>>> input = mindspore.tensor([[[ 1,  2,  3],
...     vpn free   免费的vpn梯子         vpn梯子              [ 5,  6,  7],
...    vpn梯子 免费                         [10, vpn梯子 免费 11, 12]],
...                     vpn free   vpn梯子 免费     [[ 1, vpn梯子  2,  3],
...                       vpn梯子 免费   vpn梯子 免费    [ 5,  6,  vpn梯子 免费 vpn free 7],
...             vpn永久免费梯子         免费的vpn梯子        [10, 11, 12]]])
>>> vpn free mindspore.ops.triu(input)
Tensor(shape=[2, 3, 3], dtype=Int64, value=
[[[ 1,  2,  3],
  [ 0,  6,  7],
  [ vpn梯子 0,  0, 12]],
 [[ 1,  2,  3],
  [ 0,  6,  7],
  [ 0,  0, 12]]])