Python tf.nest.flatten用法及代碼示例

從給定結構返回一個平麵列表。

用法

tf.nest.flatten(
    structure, expand_composites=False
)

有關結構的定義，請參閱tf.nest。

如果結構是原子，則返回 single-item 列表：[結構]。

這與 nest.pack_sequence_as 方法相反，該方法接受一個扁平列表並將其重新打包到嵌套結構中。

在 dict 實例的情況下，序列由值組成，按鍵排序以確保確定性行為。對於 OrderedDict 實例也是如此：它們的序列順序被忽略，而是使用鍵的排序順序。 nest.pack_sequence_as 遵循相同的約定。這會在 dicts 和 OrderedDicts 被展平後正確地重新打包它們，並且還允許展平 OrderedDict 然後使用相應的普通 dict 重新打包它，反之亦然。具有不可排序鍵的字典不能被展平。

在此函數運行時，用戶不得修改嵌套中使用的任何集合。

例子：

1. Python dict(按鍵排序)：

dict = { "key3":"value3", "key1":"value1", "key2":"value2" }
  tf.nest.flatten(dict)
    ['value1', 'value2', 'value3']

1. 對於嵌套的 python 元組：

tuple = ((1.0, 2.0), (3.0, 4.0, 5.0), 6.0)
  tf.nest.flatten(tuple)
        [1.0, 2.0, 3.0, 4.0, 5.0, 6.0]

1. 對於字典的嵌套字典：

dict = { "key3":{"c":(1.0, 2.0), "a":(3.0)},
  "key1":{"m":"val1", "g":"val2"} }
  tf.nest.flatten(dict)
    ['val2', 'val1', 3.0, 1.0, 2.0]

1. Numpy 數組(不會展平)：

array = np.array([[1, 2], [3, 4]])
  tf.nest.flatten(array)
        [array([[1, 2],
                [3, 4]])]

1. tf.Tensor(不會變平)：

tensor = tf.constant([[1., 2., 3.], [4., 5., 6.], [7., 8., 9.]])
  tf.nest.flatten(tensor)
        [<tf.Tensor:shape=(3, 3), dtype=float32, numpy=
          array([[1., 2., 3.],
                 [4., 5., 6.],
                 [7., 8., 9.]], dtype=float32)>]

1. tf.RaggedTensor ：這是一個複合張量，它的表示由 'values' 的扁平化列表和 'row_splits' 的列表組成，它們指示如何將扁平化列表分割成不同的行。有關 tf.RaggedTensor 的更多詳細信息，請訪問 https://www.tensorflow.org/api_docs/python/tf/RaggedTensor。

使用 expand_composites=False ，我們隻需按原樣返回 RaggedTensor。

tensor = tf.ragged.constant([[3, 1, 4, 1], [], [5, 9, 2]])
  tf.nest.flatten(tensor, expand_composites=False)
    [<tf.RaggedTensor [[3, 1, 4, 1], [], [5, 9, 2]]>]

使用 expand_composites=True ，我們返回組成 RaggedTensor 表示的組件張量(值和 row_splits 張量)

tensor = tf.ragged.constant([[3, 1, 4, 1], [], [5, 9, 2]])
  tf.nest.flatten(tensor, expand_composites=True)
    [<tf.Tensor:shape=(7,), dtype=int32, numpy=array([3, 1, 4, 1, 5, 9, 2],
                                                      dtype=int32)>,
     <tf.Tensor:shape=(4,), dtype=int64, numpy=array([0, 4, 4, 7])>]