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

Python tf.data.experimental.Counter用法及代码示例


创建一个 Dataset，它从 start 开始计数，步长为 step 。

用法

tf.data.experimental.Counter(
    start=0, step=1, dtype=tf.dtypes.int64
)

参数

  • start (可选。)计数器的起始值。默认为 0。
  • step (可选。)计数器的步长。默认为 1。
  • dtype (可选。)计数器元素的数据类型。默认为 tf.int64 。

返回

  • 标量 dtype 元素的 Dataset。

不像tf.data.Dataset.range 会在某个结束数字处停止，Counter 将无限期地产生元素。

dataset = tf.data.experimental.Counter().take(5)
list(dataset.as_numpy_iterator())
[0, 1, 2, 3, 4]
dataset.element_spec
TensorSpec(shape=(), dtype=tf.int64, name=None)
dataset = tf.data.experimental.Counter(dtype=tf.int32)
dataset.element_spec
TensorSpec(shape=(), dtype=tf.int32, name=None)
dataset = tf.data.experimental.Counter(start=2).take(5)
list(dataset.as_numpy_iterator())
[2, 3, 4, 5, 6]
dataset = tf.data.experimental.Counter(start=2, step=5).take(5)
list(dataset.as_numpy_iterator())
[2, 7, 12, 17, 22]
dataset = tf.data.experimental.Counter(start=10, step=-1).take(5)
list(dataset.as_numpy_iterator())
[10, 9, 8, 7, 6]

相关用法


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