View source on GitHub |
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
)
Raises | |
---|---|
ValueError
|
If the tensors do not match dtypes or shapes. |