ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

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.