Module: tf_agents.networks

View source on GitHub

Networks Module.


actor_distribution_network module: Sample Keras actor network that generates distributions.

actor_distribution_rnn_network module: Sample Keras actor network with LSTM cells that generates distributions.

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

categorical_projection_network module: Project inputs to a categorical distribution object.

dynamic_unroll_layer module: Tensorflow RL Agent RNN utilities.

encoding_network module: Keras Encoding Network.

expand_dims_layer module: Keras layer performing the equivalent of tf.expand_dims.

lstm_encoding_network module: Keras LSTM Encoding Network.

network module: Base extension to Keras network to simplify copy operations.

normal_projection_network module: Project inputs to a normal distribution object.

q_network module: Sample Keras networks for DQN.

q_rnn_network module: Sample recurrent Keras network for DQN.

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

utils module: Network utilities.

value_network module: Sample Keras Value Network.

value_rnn_network module: Sample Keras Value Network with LSTM cells .