tf_agents.utils.nest_utils.spec_shape

View source on GitHub