为什么LSTM的输出维度是二维或三维,而LSTM的输入维度始终是三维?
回答:
LSTM输入的维度必须用(样本数, 单元数, 特征数)来描述。当任务是多对多并且使用TimeDistributed层时,LSTM输出的维度必须是三维,因为return_sequence
必须设置为true
,这意味着你获得的是所有隐藏状态的值,而不仅仅是默认的最后一个值。
为什么LSTM的输出维度是二维或三维,而LSTM的输入维度始终是三维?
回答:
LSTM输入的维度必须用(样本数, 单元数, 特征数)来描述。当任务是多对多并且使用TimeDistributed层时,LSTM输出的维度必须是三维,因为return_sequence
必须设置为true
,这意味着你获得的是所有隐藏状态的值,而不仅仅是默认的最后一个值。