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

tf_agents.utils.nest_utils.assert_tensors_matching_dtypes_and_shapes

Checks if tensors have matching dtypes and shapes.

tensors_1 A nest of tensor objects.
tensors_2 A nest of tensor objects.
caller The object calling assert....
tensors_1_name (str) Name to use for tensors_1 in case of an error.
tensors_2_name (str) Name to use for tensors_2 in case of an error.

ValueError If the tensors do not match dtypes or shapes.