Function map_structure

python/paddle/utils/layers_utils.py:254–262

Apply `func` to each entry in `structure` and return a new structure.

(
    func: Callable[[_T], _U], *structure: NestedStructure[_T]
)

def map_structure(
    func: Callable[[_T], _U], *structure: NestedStructure[_T]
) -> NestedStructure[_U]:
    """
    Apply `func` to each entry in `structure` and return a new structure.
    """
    flat_structure = [flatten(s) for s in structure]
    entries = zip(*flat_structure)
    return pack_sequence_as(structure[0], [func(*x) for x in entries])


def hold_mutable_vars(structure):