0
点赞
收藏
分享

微信扫一扫

记录一个torch.tensor的语法,用于mask挑选tensor中的值

就是耍帅 2022-03-22 阅读 44
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 条评论