tf.keras.tree.map_structure

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)

func A callable that accepts as many arguments as there are structures.
*structures Arbitrarily nested structures of the same layout.

A new structure with the same layout as the given ones.