Maps func through given structures.
tf.keras.tree.map_structure(
func, *structures
)
structure = [[1], [2], [3]] keras.tree.map_structure(lambda v: v**2, structure) [[1], [4], [9]] keras.tree.map_structure(lambda x, y: x * y, structure, structure) [[1], [4], [9]]
Foo = collections.namedtuple('Foo', ['a', 'b'])
structure = Foo(a=1, b=2)
keras.tree.map_structure(lambda v: v * 2, structure)
Foo(a=2, b=4)| Args | |
|---|---|
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 the given ones. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/keras/tree/map_structure