解决RNN模型中'For unbatched 2-D input, hx should also be 2-D but got 3-D tensor'错误
这个问题是因为在使用RNN模型时,输入的数据是2维的,但是隐藏状态hx被错误地传递为3维张量。解决方法是将hx改为2维张量,可以使用squeeze()函数来压缩维度,例如:
hx = hx.squeeze(0)
这将消除hx张量的第一维,将其转换为2维张量。然后可以将hx传递给RNN模型。
原文地址: http://www.cveoy.top/t/topic/oli1 著作权归作者所有。请勿转载和采集!