View source on GitHub |
Networks Module.
Modules
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.
categorical_projection_network
module: Project inputs to a categorical distribution object.
categorical_q_network
module: A Q-network for categorical DQN.
encoding_network
module: Keras Encoding Network.
expand_dims_layer
module: Keras layer performing the equivalent of tf.expand_dims.
layer_utils
module: Network layer utilities.
lstm_encoding_network
module: Keras LSTM Encoding Network.
mask_splitter_network
module: Wrapper network that handles action constraint portion of the observation.
nest_map
module: Network layer that allows mapping multiple inputs.
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
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 .
Classes
class NestMap
: The NestMap
network processes nested inputs via nested layers.
class Network
: A class used to represent networks used by TF-Agents policies and agents.
class Sequential
: The Sequential Network represents a sequence of Keras layers.
Functions
NestFlatten(...)
: Returns a Keras layer that takes a nest of inputs, and returns a list.