Module: tf_agents.keras_layers.rnn_wrapper

Stay organized with collections Save and categorize content based on your preferences.

Wrapper for tf.keras.layers.RNN subclasses.

Classes

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