在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