Python tf.data.experimental.RandomDataset.take用法及代碼示例

用法

take(
    count, name=None
)

count 一個 tf.int64 標量 tf.Tensor ，表示該數據集的元素數，應用於生成新數據集。如果count 為-1，或者count 大於此數據集的大小，則新數據集將包含此數據集的所有元素。
name (可選。) tf.data 操作的名稱。

從該數據集中創建一個最多包含 count 元素的 Dataset。

dataset = tf.data.Dataset.range(10)
dataset = dataset.take(3)
list(dataset.as_numpy_iterator())
[0, 1, 2]

相關用法

注：本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.data.experimental.RandomDataset.take。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。