2021年12月5日
Pytorch中的掩码：dtype=torch.uint8
摘要： 在pytorch中，dtype=uint8的数据类型往往可以用作掩码，0表示舍弃对应项，1表示选取对应项。通过设置不同的0或1的值，对另外的tensor进行选择性选取： 例如： t = torch.rand(4，2) """ tensor([[0.5492, 0.2083], [0.3635, 0. 阅读全文
posted @ 2021-12-05 20:07 cold_moon的笔记 阅读(879) 评论(0) 推荐(0) 编辑