tf_agents.utils.nest_utils.matching_dtypes_and_inner_shapes

View source on GitHub

Returns True if tensors and specs have matching dtypes and inner shapes.

tensors A nest of tensor objects.
specs A nest of tf.TypeSpec objects.
allow_extra_fields If True, then tensors may contain more keys or list fields than strictly required by specs.

A python bool.