Module: tf_agents.keras_layers

Keras Layers Module.


bias_layer module: Keras layer mirroring tf.contrib.layers.bias_add.

dynamic_unroll_layer module: Tensorflow RL Agent RNN utilities.

inner_reshape module: Keras layer to reshape inner dimensions (keeping outer dimensions the same).

rnn_wrapper module: Wrapper for tf.keras.layers.RNN subclasses.

sequential_layer module: Keras layer to replace the Sequential Model object.


class BiasLayer: Keras layer that only adds a bias to the input.

class DynamicUnroll: Process a history of sequences that are concatenated without padding.

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

class SequentialLayer: The SequentialLayer represents a sequence of Keras layers.


InnerReshape(...): Returns a Keras layer that reshapes the inner dimensions of tensors.