RuntimeError: expected scalar type Double but found Float
将out = net(A_wave, val_input)改为out = net(A_wave.to(torch.float32), val_input)错误就消失了。
原因:tensor的数据类型不正确
解决: 将数据类型转为float32
并不是因为他expected Double你就要弄成Double
查了一些博文说
tensor.to(torch.float32)
但其实这里面的tensor指的是tensor类型的数据
在你自己的代码上他可能是x可能是data
这个要你自己多尝试
我的是A_wave.to(torch.float32).
内容转自https://blog.csdn.net/ttdxtt/article/details/112347283