为什么输出LSTM层的维度既可以是2维也可以是3维?

Why the dimension of output LSTM layer can be either 2 dimension or 3 dimension?

为什么 LSTM 输出的维度是 2 或 3,而 LSTM 输入的维度总是 3 维?

LSTM 输入的维度必须用 (sample_number, unit_number, feature_number) 来描述。当任务为 many-to-many 并且您使用的是 TimeDistributed 层时,LSTM 输出的维度必须为 3,因为 return_sequence 必须为 true,这意味着您将获得所有隐藏状态值而不是只有最后一个值作为默认值。