Google I/O is a wrap! Catch up on TensorFlow sessions View sessions

tf_agents.utils.nest_utils.assert_tensors_matching_dtypes_and_shapes

Checks if tensors have matching dtypes and shapes.

tf_agents.utils.nest_utils.assert_tensors_matching_dtypes_and_shapes(
    tensors_1, tensors_2, caller, tensors_1_name, tensors_2_name
)

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.