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