pytorch用type转换数据类型报Expected type ‘Union[type, str]‘, got ‘dtype‘ instead警告

pytorch用type转换数据类型报Expected type ‘Union[type, str]‘, got ‘dtype‘ instead警告_第1张图片
这是数据类型不对,torch.float32是dtype型的,我们要输一个type型的
pytorch用type转换数据类型报Expected type ‘Union[type, str]‘, got ‘dtype‘ instead警告_第2张图片
所以把torch.float32改成torch.FloatTensor就可以了,不警告了
在这里插入图片描述
类似的 torch.bool可以改成torch.BoolTensor

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