Have a question? Connect with the community at the TensorFlow Forum Visit Forum

Module: tf_agents.networks.network

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 module given input_spec; return output_spec.

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