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.