tf_agents.specs.tensor_spec.with_dtype

Updates dtypes of all specs in the input spec.

specs Nested list/tuple/dict of TensorSpecs/ArraySpecs, describing the shape of tensors.
dtype dtype to update the specs to.

Nested TensorSpecs with the udpated dtype.