记录一个torch.tensor的语法，用于mask挑选tensor中的值-CFANZ编程社区

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

就是耍帅

阅读 44

2022-03-22

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:
在这里插入图片描述

相关推荐
yongxinz
 Pytorch中torch.Tensor和torch.tensor()以及其他Tensor类型的区别
yongxinz 53 0 0
郝春妮
 torch.Tensor和torch.tensor有什么区别？
郝春妮 81 0 0
凯约
 Pytorch实用教程： torch.tensor()的用法
凯约 7 0 0
耳一文
 每日学习记录01：numpy和torch.tensor中对张量的基本操作
耳一文 9 0 0
老罗话编程
 torch.Tensor行向量转为列向量（unsqueeze）
老罗话编程 66 0 0
犹大之窗
 两个tensor作为另一个tensor的索引
犹大之窗 61 0 0
幸甚至哉歌以咏志
 Pytorch学习之：索引出一个 tensor 中的最大值及其所在的位置 torch.argmax
幸甚至哉歌以咏志 357 0 0
惠特曼
 CUDA中的Tensor
惠特曼 49 0 0
辰鑫chenxin
 机器学习笔记之list, numpy.array, torch.Tensor 格式相互转化
辰鑫chenxin 82 0 0
12a597c01003
 【PyTorch】利用torch.cat()实现Tensor的拼接
12a597c01003 171 0 0

精彩评论（0）