Attend the Women in ML Symposium on December 7 Register now

tf_agents.specs.tensor_spec.with_dtype

Stay organized with collections Save and categorize content based on your preferences.

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.