当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python tf.compat.v1.data.experimental.RandomDataset.apply用法及代码示例


用法

apply(
    transformation_func
)

参数

  • transformation_func 一个接受一个 Dataset 参数并返回一个 Dataset 的函数。

返回

  • Dataset 通过将此数据集应用transformation_func 返回的Dataset

将转换函数应用于此数据集。

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]

相关用法


注:本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.compat.v1.data.experimental.RandomDataset.apply。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。