在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