mindspore.mint.tril

mindspore.mint.tril(input, diagonal=0)[源代码]

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

参数:
  • input (Tensor) - 输入tensor,其秩至少为2。

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

返回:

Tensor

支持平台:

Ascend

样例:

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