Expected object of scalar type Float but got scalar type Double for argument

2022-09-02 20:28:13 浏览数 (2)

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

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

代码语言:javascript复制
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

分类:

0 人点赞