Keras 3 API 文档 / 层 API / 循环层 / 堆叠 RNN 单元层

堆叠 RNN 单元层

[源代码]

StackedRNNCells

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

包装器,允许 RNN 单元堆栈表现为单个单元。

用于实现高效的堆叠 RNN。

参数

  • cells: RNN 单元实例列表。

示例

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)