La solución es convertir los datos utilizados:
Resulta que:
prediction, h_state = rnn(x, h_state)
Conversión de tipo plus:
x = torch.tensor(x, dtype=torch.float32)
prediction, h_state = rnn(x, h_state)
RuntimeError: Objeto esperado de tipo escalar Doble pero tiene tipo escalar Flotante para argumento # 2 'mat2'
Supongo que te gusta
Origin blog.csdn.net/comli_cn/article/details/104609123
Recomendado
Clasificación