这是数据类型不对,torch.float32是dtype型的,我们要输一个type型的
所以把torch.float32改成torch.FloatTensor就可以了,不警告了
类似的 torch.bool可以改成torch.BoolTensor

这是数据类型不对,torch.float32是dtype型的,我们要输一个type型的
所以把torch.float32改成torch.FloatTensor就可以了,不警告了
类似的 torch.bool可以改成torch.BoolTensor