tf.contrib.framework.nest.flatten_up_to

Разворачивает input_tree до shallow_tree.

tf.contrib.framework.nest.flatten_up_to(
    shallow_tree, input_tree, check_types=True, expand_composites=False,
    check_subtrees_length=True
)

Любая дальнейшая глубина структуры в input_tree сохраняется как элементы в частично развернутом выводе.

Если shallow_tree и input_tree не являются последовательностями, это возвращает список из одного элемента: [input_tree].

Случай использования:

Иногда нам может потребоваться частично развернуть вложенную последовательность, сохранив часть вложенной структуры. Мы достигаем этого, указав мелкую структуру, shallow_tree, до которой мы хотим развернуть.

Входные данные, input_tree, можно рассматривать как имеющие ту же структуру, что и shallow_tree, но с узлами-листьями, которые сами являются древовидными структурами.

Примеры:

input_tree = [[[2, 2], [3, 3]], [[4, 9], [5, 5]]]
shallow_tree = [[True, True], [False, True]]

flattened_input_tree = flatten_up_to(shallow_tree, input_tree)
flattened_shallow_tree = flatten_up_to(shallow_tree, shallow_tree)

# Output is:
# [[2, 2], [3, 3], [4, 9], [5, 5]]
# [True, True, False, True]

input_tree = [[('a', 1), [('b', 2), [('c', 3), [('d', 4)]]]]]
shallow_tree = [['level_1', ['level_2', ['level_3', ['level_4']]]]]

input_tree_flattened_as_shallow_tree = flatten_up_to(shallow_tree, input_tree)
input_tree_flattened = flatten(input_tree)

# Output is:
# [('a', 1), ('b', 2), ('c', 3), ('d', 4)]
# ['a', 1, 'b', 2, 'c', 3, 'd', 4]

Случаи с не-последовательностями:

flatten_up_to(0, 0)  # Output: [0]
flatten_up_to(0, [0, 1, 2])  # Output: [[0, 1, 2]]
flatten_up_to([0, 1, 2], 0)  # Output: TypeError
flatten_up_to([0, 1, 2], [0, 1, 2])  # Output: [0, 1, 2]

Случай неполного поддерева:

shallow_tree = ["a", "b"]
input_tree = ["c", ["d", "e"], "f"]
flattened = flatten_up_to(shallow_tree, input_tree,
  check_subtrees_length=False)

# Output is:
# ["c", ["d", "e"]]