Google I/O returns May 18-20! Reserve space and build your schedule Register now

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.