Add map_structure_with_paths to util.nest.
map_structure_with_paths is like map_structure, except that the string path of each leaf in the structure is passed as the first argument to the function. This can be used for example to create variables whose names are derived from the string path. PiperOrigin-RevId: 206776927
Loading
Please sign in to comment