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