torch中,tensor数值类型变成bool值

直接使用下面的语句会报错:‘Tensor’ object has no attribute ‘bool’

a = torch.ones(2,2)
a.bool()

应该改成:

a = torch.ones(2,2)
a.byte()

你可能感兴趣的:(深度学习,python,pytorch,pytorch,python)