Maps func through given structures.
tf.keras.tree.map_structure(
func, *structures
)
Examples:
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.
|
View source on GitHub