![]() |
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
)
Args | |
---|---|
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. |
Raises | |
---|---|
ValueError
|
If the tensors do not match dtypes or shapes. |