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