ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

Module: tf_agents.networks.categorical_projection_network

Project inputs to a categorical distribution object.

Classes

class CategoricalProjectionNetwork: Generates a tfp.distribution.Categorical by predicting logits.