tf.keras.tree.map_structure

Maps func through given structures.

tf.keras.tree.map_structure(
    func, *structures, none_is_leaf=True
)

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.
none_is_leaf	If True, func will be called on None leaves. If False, None values are not passed to func and are returned in the output directly.

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

Raises
TypeError	If structures is empty or func is not callable.
ValueError	If there is more than one items in structures and some of the nested structures don't match according to the rules of assert_same_structure.