TensorFlow 1 version | View source on GitHub |
Wrapper allowing a stack of RNN cells to behave as a single cell.
Inherits From: Layer
tf.keras.layers.StackedRNNCells(
cells, **kwargs
)
Used to implement efficient stacked RNNs.
Arguments | |
---|---|
cells
|
List of RNN cell instances. |
Examples:
cells = [
keras.layers.LSTMCell(output_dim),
keras.layers.LSTMCell(output_dim),
keras.layers.LSTMCell(output_dim),
]
inputs = keras.Input((timesteps, input_dim))
x = keras.layers.RNN(cells)(inputs)
Attributes | |
---|---|
output_size
|
|
state_size
|
Methods
get_initial_state
get_initial_state(
inputs=None, batch_size=None, dtype=None
)