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

tf_agents.distributions.utils.assert_specs_are_compatible

Checks that the output of network.create_variables matches a spec.

network_output_spec The output of network.create_variables.
spec The spec we are matching to.
message_prefix The message prefix for error messages, used when the specs don't match.

ValueError If the specs don't match.