ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

Module: tf_agents.keras_layers.rnn_wrapper

Wrapper for tf.keras.layers.RNN subclasses.

Classes

class RNNWrapper: Wraps a Keras RNN/LSTM/GRU layer to make network state more consistent.