|View source on GitHub|
Base extension to Keras network to simplify copy operations.
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.
create_variables(...): Create variables in
get_state_spec(...): Extracts the state spec from a layer.