tf.keras.tree.map_structure_up_to | TensorFlow v2.16.1 (original) (raw)
tf.keras.tree.map_structure_up_to
Stay organized with collections Save and categorize content based on your preferences.
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']]
| Args | |
|---|---|
| shallow_structure | A structure with layout common to all structures. |
| func | A callable that accepts as many arguments as there are structures. |
| *structures | Arbitrarily nested structures of the same layout. |
| Returns |
|---|
| A new structure with the same layout as shallow_structure. |
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.
Last updated 2024-06-07 UTC.