解决方法是将用到的数据进行类型转换：
原来：
prediction, h_state = rnn(x, h_state)
加上类型转换：
x = torch.tensor(x, dtype=torch.float32)
prediction, h_state = rnn(x, h_state)
RuntimeError: Expected object of scalar type Double but got scalar type Float for argument #2 'mat2'
最新推荐文章于 2022-06-27 20:40:58 发布
本文介绍在PyTorch框架中，如何通过数据类型转换解决运行时错误。具体方法是在输入数据前使用torch.tensor函数，并指定dtype参数为torch.float32，确保数据类型与模型兼容。
扫一扫
6835
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
