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]