RuntimeError: Expected object of scalar type Double but got scalar type Float for argument #2 ‘mat2’ in call to _th_mm
在调用函数时,在函数的下面代码出错
**原因:**类型问题
方法:本人是在调用此函数之前修改成w.double(),但是出问题的是下面这一句,因此应该修改为下面这样:
xw = torch.matmul(inputs.to(device), w.double()).to(device) 即可
RuntimeError:Expected object of scalar type Double
最新推荐文章于 2023-07-04 15:40:39 发布