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]]])