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

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


用法

take(
    count, name=None
)

参数

  • count 一个 tf.int64 标量 tf.Tensor ，表示该数据集的元素数，应用于生成新数据集。如果count 为-1，或者count 大于此数据集的大小，则新数据集将包含此数据集的所有元素。
  • name (可选。) tf.data 操作的名称。

返回

  • Dataset 一个Dataset。

从该数据集中创建一个最多包含 count 元素的 Dataset。

dataset = tf.data.Dataset.range(10)
dataset = dataset.take(3)
list(dataset.as_numpy_iterator())
[0, 1, 2]

相关用法


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