Keras 3 API 文件 / 層 API / 循環層 / 堆疊 RNN cell 層

堆疊 RNN cell 層

[原始碼]

StackedRNNCells 類別

keras.layers.StackedRNNCells(cells, **kwargs)

封裝器,允許堆疊的 RNN cells 作為單一 cell 運作。

用於實作高效堆疊 RNNs。

參數

  • cells:RNN cell 實例的列表。

範例

batch_size = 3
sentence_length = 5
num_features = 2
new_shape = (batch_size, sentence_length, num_features)
x = np.reshape(np.arange(30), new_shape)

rnn_cells = [keras.layers.LSTMCell(128) for _ in range(2)]
stacked_lstm = keras.layers.StackedRNNCells(rnn_cells)
lstm_layer = keras.layers.RNN(stacked_lstm)

result = lstm_layer(x)