我相信错误是因为我有一个LSTM层。我如何修改代码,使它将工作正常?任何帮助?
py49o6xq1#
将LSTM层移出顺序层。LSTM返回output, (hn, cn)的元组,其中hn, cn是最后的隐藏状态。例如,您的init函数将包含如下内容
LSTM
output, (hn, cn)
hn, cn
class module(nn.Module): def __init__(self): super(nn.Module, self).__init__() self.lstm = nn.LSTM(...) self.seq = nn.Sequential(...)
转发函数将为
def forward(self, x): lstm_out= self.lstm(x) out = self.seq(lstm_out[0]) return out
1条答案
按热度按时间py49o6xq1#
将
LSTM
层移出顺序层。LSTM返回output, (hn, cn)
的元组,其中hn, cn
是最后的隐藏状态。例如,您的init函数将包含如下内容
转发函数将为