tf.nest.map_structure | TensorFlow v2.16.1 (original) (raw)

tf.nest.map_structure

Stay organized with collections Save and categorize content based on your preferences.

Creates a new structure by applying func to each atom in structure.

View aliases

Compat aliases for migration

tf.compat.v1.nest.map_structure

tf.nest.map_structure(
    func, *structure, **kwargs
)

Used in the notebooks

Used in the guide	Used in the tutorials
Import a JAX model using JAX2TF Distributed training with Core APIs and DTensor	Distributed training with DTensors Substantial Undocumented Infection Facilitates the Rapid Dissemination of Novel Coronavirus (SARS-CoV2) Use TFF optimizers in custom iterative process Variational Inference on Probabilistic Graphical Models with Joint Distributions Composing Learning Algorithms

Refer to tf.nestfor the definition of a structure.

Applies func(x[0], x[1], ...) where x[i] enumerates all atoms instructure[i]. All items in structure must have the same arity, and the return value will contain results with the same structure layout.

Examples:

A single Python dict:

a = {"hello": 24, "world": 76} tf.nest.map_structure(lambda p: p * 2, a) {'hello': 48, 'world': 152}

Multiple Python dictionaries:

d1 = {"hello": 24, "world": 76} d2 = {"hello": 36, "world": 14} tf.nest.map_structure(lambda p1, p2: p1 + p2, d1, d2) {'hello': 60, 'world': 90}

A single Python list:

a = [24, 76, "ab"] tf.nest.map_structure(lambda p: p * 2, a) [48, 152, 'abab']

Scalars:

tf.nest.map_structure(lambda x, y: x + y, 3, 4) 7

Empty structures:

tf.nest.map_structure(lambda x: x + 1, ()) ()

Check the types of iterables:

s1 = (((1, 2), 3), 4, (5, 6)) s1_list = [[[1, 2], 3], 4, [5, 6]] tf.nest.map_structure(lambda x, y: None, s1, s1_list) Traceback (most recent call last): `` TypeError: The two structures don't have the same nested structure

Type check is set to False:

s1 = (((1, 2), 3), 4, (5, 6)) s1_list = [[[1, 2], 3], 4, [5, 6]] tf.nest.map_structure(lambda x, y: None, s1, s1_list, check_types=False) (((None, None), None), None, (None, None))

Args
func	A callable that accepts as many arguments as there are structures.
*structure	atom or nested structure.
**kwargs	Valid keyword args are: check_types: If set to True (default) the types of iterables within the structures have to be same (e.g. map_structure(func, [1], (1,))raises a TypeError exception). To allow this set this argument toFalse. Note that namedtuples with identical name and fields are always considered to have the same shallow structure. expand_composites: If set to True, then composite tensors such astf.sparse.SparseTensor and tf.RaggedTensor are expanded into their component tensors. If False (the default), then composite tensors are not expanded.

Returns
A new structure with the same arity as structure[0], whose atoms correspond to func(x[0], x[1], ...) where x[i] is the atom in the corresponding location in structure[i]. If there are different structure types and check_types is False the structure types of the first structure will be used.

Raises
TypeError	If func is not callable or if the structures do not match each other by depth tree.
ValueError	If no structure is provided or if the structures do not match each other by type.
ValueError	If wrong keyword arguments are provided.