Save the date! Google I/O returns May 18-20 Register now

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.

squashed_outer_wrapper module: SquashedOuterWrapper Keras Layer.


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.

class SquashedOuterWrapper: Squash the outer dimensions of input tensors; unsquash outputs.


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