【PyTorch】Chapter1：expected dtype Double but got dtype Float

1、简介

最近跟着《Dive into DL PyTorch》一书的代码写，出现了double类型的数据和float类型的数据之间相互冲突的问题，有时需要double类型但输入的是float类型，有时又需要float类型却得到double类型，因此代码运行错误。在书中，自己定义net，loss，optimizer，weight，bias等等，这可以直接指定数据类型，但是每次到了使用库函数时，比如书中说的简约版代码时，总是会报数据类型的错误，而且都是weight和bias这两个parameters。这两个是构造net的时候自动产生，因此无法直接改动其数据类型。

2、解决

每次加上：

torch.set_default_tensor_type(torch.DoubleTensor)

或者

torch.set_default_tensor_type(torch.FloatTensor)

直接设置创建的tensor类型默认为Double，如果不设置的话自动默认为float类型。