参考来自这个github问题。
当错误出现时
RuntimeError: Expected object of scalar type Float but got scalar typeDouble for argument #4 'mat1'，您将需要使用该
.float()函数，因为它说的是
Expected object ofscalar type Float。
因此，解决方案更改
y_pred = model(X_trainTensor)为
y_pred =model(X_trainTensor.float())。
同样，当您遇到另一个错误时
loss = loss_fn(y_pred,y_trainTensor)，
y_trainTensor.long()由于错误消息中显示，因此您也需要
Expected object ofscalar type Long。
您也可以
model.double()按照@Paddy的建议进行操作。
