kk = torch.tensor([[4,5],[7,8]])
x = torch.zeros(2,2)
y = torch.tensor([[1,0],[0,1]])
print([x.byte() |y.byte()])
print(kk[x.byte() |y.byte()])
outputs:
0
点赞
收藏
分享
记录一个torch.tensor的语法，用于mask挑选tensor中的值
0 条评论
kk = torch.tensor([[4,5],[7,8]])
x = torch.zeros(2,2)
y = torch.tensor([[1,0],[0,1]])
print([x.byte() |y.byte()])
print(kk[x.byte() |y.byte()])
outputs:
