RuntimeError: Expected object of scalar type Float but got scalar type Double for argument #4 'mat1'的解决方法

在pytorch中float32为float类型,而float64则为double类型,注意tensor的数据类型。

可以通过指定数据类型来获得所需要的类型数据。

torch.zeros([self.batch_size, self.num_layers, self.hidden_size], dtype=torch.float32) # float
torch.zeros([self.batch_size, self.num_layers, self.hidden_size], dtype=torch.float64) # double
原文地址:https://www.cnblogs.com/wangling1820/p/12533473.html