Pytorch:module 'torch' has no attribute 'bool'
这个应该是有些版本的Pytorch会遇到这个问题,我用0.4.0版本测试发现torch.bool是有的,但是1.0.0确实没有了,我看了一篇文章:
来源:http://www.sohu.com/a/332756215_473283
Pytorch 1.2.0 来了!
比较操作返回的数据类型由 torch.uint8 变为 torch.bool
1>>> torch.tensor([1, 2, 3]) < torch.tensor([3, 1, 2])
2tensor([1, 0, 0], dtype=torch.uint8)
所以应该猜测是1.2.0之前的版本会有这个问题。
解决办法就是把torch.bool换成torch.uint8