tf_agents.distributions.utils.assert_specs_are_compatible

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

tf_agents.distributions.utils.assert_specs_are_compatible(
    network_output_spec: tf_agents.typing.types.NestedTensorSpec,
    spec: tf_agents.typing.types.NestedTensorSpec,
    message_prefix: str
)

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.