tf_agents.utils.nest_utils.fast_map_structure_flatten

View source on GitHub