tf.keras.tree.map_structure
Применяет func к заданным структурам.
tf.keras.tree.map_structure(
func, *structures
)
Примеры:
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 | Вызываемый объект, принимающий столько аргументов, сколько имеется структур. |
*structures | Произвольно вложенные структуры с одинаковой компоновкой. |
| Возвращаемое значение | |
|---|---|
| Новая структура с той же компоновкой, что и у заданных структур. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/keras/tree/map_structure