RuntimeError: expected scalar type Double but found Float

网络传入的类型错误，我一开始是expected scalar type Floatbut found long

train_tensor = train_tensor.to(torch.float32)

然后就报

RuntimeError: expected scalar type Double but found Float

然后直接网络哪里修改

然后就好了开始训练