Module: tf_agents.networks.network

Base extension to Keras network to simplify copy operations.

Classes

class DistributionNetwork: Base class for networks which generate Distributions as their output.

class Network: A class used to represent networks used by TF-Agents policies and agents.

Functions

create_variables(...): Create variables in module given input_spec; return output_spec.

get_state_spec(...): Extracts the state spec from a layer.