View source on GitHub
|
Flattens a possibly nested structure into a list.
tf.keras.tree.flatten(
structure
)
In the case of dict instances, the sequence consists of the values,
sorted by key to ensure deterministic behavior. However, instances of
collections.OrderedDict are handled differently: their sequence order is
used instead of the sorted keys. The same convention is followed in
pack_sequence_as. This correctly unflattens dicts and OrderedDict after
they have been flattened, or vice-versa.
Dictionaries with non-sortable keys are not supported.
Examples:
keras.tree.flatten([[1, 2, 3], [4, [5], [[6]]]])[1, 2, 3, 4, 5, 6]keras.tree.flatten(None)[None]keras.tree.flatten(1)[1]keras.tree.flatten({100: 'world!', 6: 'Hello'})['Hello', 'world!']
Args | |
|---|---|
structure
|
An arbitrarily nested structure. |
Returns | |
|---|---|
A list, the flattened version of the input structure.
|