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


Python tf.compat.v1.data.TFRecordDataset.skip用法及代码示例


用法

skip(
    count, name=None
)

参数

  • count 一个 tf.int64 标量 tf.Tensor ,表示应跳过此数据集以形成新数据集的元素数。如果count 大于此数据集的大小,则新数据集将不包含任何元素。如果 count 为 -1,则跳过整个数据集。
  • name (可选。) tf.data 操作的名称。

返回

  • Dataset 一个Dataset

创建一个 Dataset 跳过此数据集中的 count 元素。

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

相关用法


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