tf.contrib.framework.nest.get_traverse_shallow_structure
Генерирует плоскую структуру из traverse_fn и structure.
tf.contrib.framework.nest.get_traverse_shallow_structure(
traverse_fn, structure, expand_composites=False
)
traverse_fn должен принимать любой возможный поддерево structure и возвращать структуру глубиной 1, содержащую True или False значения, описывающие, какие поддеревья верхнего уровня можно пройти. Он также может возвращать скалярные True или False "обход разрешен/не разрешен для всех поддеревьев".
Примеры доступны в тестовых модулях (nest_test.py).
| Аргументы | |
|---|---|
traverse_fn | Функция, принимающая подструктуру и возвращающая либо скалярное bool (следует ли обходить эту подструктуру) или плоскую структуру глубиной 1 того же типа, описывающую, какие части подструктуры обходить. |
structure | Структура для обхода. |
expand_composites | Если True, составные тензоры, такие как tf.SparseTensor и tf.RaggedTensor, расширяются до их компонентных тензоров. |
| Возвращаемое значение | |
|---|---|
Плоская структура, содержащая значения типа python bool, которые можно передать в map_structure_up_to и flatten_up_to. |
| Исключения | |
|---|---|
TypeError | если traverse_fn возвращает последовательность для не-последовательного входного значения, или структуру с глубиной больше 1 для последовательного входного значения, или если любые листовые значения в возвращаемой структуре или скаляре не являются типом bool. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/contrib/framework/nest/get_traverse_shallow_structure
