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

用法

apply(
    transformation_func
)

將轉換函數應用於此數據集。

apply 啟用自定義 Dataset 轉換的鏈接，這些轉換表示為接受一個 Dataset 參數並返回轉換後的 Dataset 的函數。

dataset = tf.data.Dataset.range(100)
def dataset_fn(ds):
  return ds.filter(lambda x:x < 5)
dataset = dataset.apply(dataset_fn)
list(dataset.as_numpy_iterator())
[0, 1, 2, 3, 4]