View source on GitHub |
Maps func
through given structures up to shallow_structure
.
tf.keras.tree.map_structure_up_to(
shallow_structure, func, *structures
)
This is a variant of map_structure
which only maps the given structures
up to shallow_structure
. All further nested components are retained as-is.
Examples:
shallow_structure = [None, None]
structure = [[1, 1], [2, 2]]
keras.tree.map_structure_up_to(shallow_structure, len, structure)
[2, 2]
shallow_structure = [None, [None, None]]
keras.tree.map_structure_up_to(shallow_structure, str, structure)
['[1, 1]', ['2', '2']]
Returns | |
---|---|
A new structure with the same layout as shallow_structure .
|