Keras RNN 中的单位参数

Question

我很难理解以下 Keras sn-p 实现的架构：

HIDDEN_DIM=256
VOCAB_SIZE=80
model = Sequential()
model.add(LSTM(units=HIDDEN_DIM, input_shape=(None, VOCAB_SIZE), return_sequences=True))
for i in range(LAYER_NUM - 1):
    model.add(LSTM(units=HIDDEN_DIM, return_sequences=True))
model.add(TimeDistributed(Dense(VOCAB_SIZE)))
model.add(Activation('softmax'))
model.compile(loss="categorical_crossentropy", optimizer="rmsprop")

具体来说，我不太了解“单位”参数。 unit参数是定义时间步的数量，还是定义每个时间步的LSTM数量（即“通道”的数量）？

Answer 1

Units 参数定义了 LSTM 中隐藏单元的数量。时间步数将由句子中的单词数定义。