Python tf.data.TFRecordDataset.skip用法及代碼示例

用法

skip(
    count, name=None
)

count 一個 tf.int64 標量 tf.Tensor ，表示應跳過此數據集以形成新數據集的元素數。如果count 大於此數據集的大小，則新數據集將不包含任何元素。如果 count 為 -1，則跳過整個數據集。
name (可選。) tf.data 操作的名稱。

創建一個 Dataset 跳過此數據集中的 count 元素。

dataset = tf.data.Dataset.range(10)
dataset = dataset.skip(7)
list(dataset.as_numpy_iterator())
[7, 8, 9]

相關用法

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