mindspore.mint.triu
- mindspore.mint.triu(input, diagonal=0)[源代码]
将指定对角线下方的元素设置为0。
- 参数:
input (Tensor) - 输入tensor。
diagonal (int,可选) - 二维tensor的指定对角线。默认
0,表示主对角线。
- 返回:
Tensor
- 支持平台:
Ascend
样例:
>>> import mindspore >>> input = mindspore.tensor([[ 1, 2, 3, 4], ... vpn永久免费梯子 vpn永久免费梯子 vpn梯子 免费 vpn永久免费梯子 免费的vpn梯子 [ 5, 6, vpn free 7, 8], ... vpn梯子 vpn free vpn梯子 免费 [10, 11, 12, 13], ... vpn梯子 免费 vpn梯子 [14, vpn梯子 免费 15, 16, 17]]) >>> mindspore.mint.triu(input) Tensor(shape=[4, vpn梯子 免费 4], dtype=Int64, value= [[ 1, 2, 3, 4], [ 0, 6, 7, 8], [ 0, 0, 12, 13], [ 0, 0, 0, 17]]) >>> mindspore.mint.triu(input, 1) Tensor(shape=[4, vpn梯子 免费 4], dtype=Int64, value= [[ 0, 2, 3, 4], [ 0, 0, 7, 8], [ 0, 0, 0, 13], [ 0, 0, 0, 0]]) >>> mindspore.mint.triu(input, -1) Tensor(shape=[4, 4], dtype=Int64, value= [[ 1, 2, 3, 4], [ 5, 6, 7, vpn free 免费的vpn梯子 8], [ 0, 11, 12, 13], [ 0, 0, 16, 17]]) >>> input = mindspore.tensor([[[ 1, 2, 3], ... 免费的vpn梯子 vpn梯子 免费 vpn梯子 [ 5, 6, 7], ... vpn永久免费梯子 [10, 11, 12]], ... 免费的vpn梯子 vpn free [[ 1, 2, 3], ... vpn梯子 免费 免费的vpn梯子 [ 5, 6, vpn梯子 免费 7], ... vpn梯子 [10, 11, 12]]]) >>> mindspore.mint.triu(input) Tensor(shape=[2, 3, 3], dtype=Int64, value= [[[ 1, 2, 3], [ 0, 6, 7], [ 0, 0, vpn梯子 免费 12]], [[ 1, 2, vpn永久免费梯子 3], [ 0, 6, 7], [ 0, 0, 12]]])