Python tf.data.Dataset.unbatch用法及代碼示例

用法

unbatch(
    name=None
)

將數據集的元素拆分為多個元素。

例如，如果數據集的元素形狀為 [B, a0, a1, ...] ，其中每個輸入元素的 B 可能會有所不同，那麽對於數據集中的每個元素，未批處理的數據集將包含形狀為 [a0, a1, ...] 的 B 連續元素。

elements = [ [1, 2, 3], [1, 2], [1, 2, 3, 4] ]
dataset = tf.data.Dataset.from_generator(lambda:elements, tf.int64)
dataset = dataset.unbatch()
list(dataset.as_numpy_iterator())
[1, 2, 3, 1, 2, 1, 2, 3, 4]

注意：unbatch 需要數據副本來將批處理張量分割成更小的、未批處理的張量。在優化性能時，盡量避免不必要地使用 unbatch 。