RuntimeError: Expected object of scalar type Double but got scalar type Float for argument #2 'mat2'

在

#线性回归矢量计算
def linreg(X,w,b):
    return torch.mm(X,w)+b

报出

RuntimeError: Expected object of scalar type Double but got scalar type Float for argument #2 'mat2'

错误。其主要原因是第二个参数本应该是double类型，但传过来的是float类型。只需要更改一下类型就行。

w = w.double()