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.