tf_agents.specs.array_spec.update_spec_shape

View source on GitHub

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