tf_agents.specs.array_spec.update_spec_dtype

View source on GitHub

Returns a copy of the given spec with the new dtype.