原文:UserWarning: indexing with dtype torch.uint8 is now deprecated, please use a dtype torch.bool instead.

内部实则是bool。 原: 可以看出,实则idxs内部为bool性质。 改后: ...

2020-05-06 16:37 0 2017 推荐指数:

查看详情

Pytorch中的掩码:dtype=torch.uint8

在pytorch中,dtype=uint8的数据类型往往可以用作掩码,0表示舍弃对应项,1表示选取对应项。通过设置不同的0或1的值,对另外的tensor进行选择性选取: 例如: ...

Mon Dec 06 04:07:00 CST 2021 0 797
 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM