tf.nest.map_structure

Применяет func к каждой записи в structure и возвращает новую структуру.

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

Применяет func(x[0], x[1], ...) , где x[i] — запись в structure[i]. Все структуры в structure должны иметь одинаковую арность, а значение возврата будет содержать результаты с одинаковой структурой.

Примеры:

1. Один Python-словарь:

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

1. Несколько Python-словарей:

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}