LSTM中的input_size=3 num_layers=1 hidden_size=12是否代表每一个LSTM层会有12个LSTM单元
时间: 2024-04-28 19:26:54 浏览: 128
多层LSTM,多层lstm与单层lstm区别,Python
5星 · 资源好评率100%
在这种情况下,每个LSTM层确实有12个LSTM单元。在LSTM中,隐藏状态的维度(也称为隐藏单元或输出数量)由hidden_size参数定义。因此,当hidden_size=12时,每个LSTM层都将包含12个LSTM单元。请注意,在LSTM中,每个LSTM单元内部有多个神经元,但是hidden_size参数代表的是每个LSTM单元的输出维度,而不是LSTM单元内部神经元的数量。
阅读全文