tf.data.experimental.Counter

Creates a Dataset that counts from start in steps of size step. (deprecated)

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

Used in the notebooks

Used in the guide	Used in the tutorials
tf.data: Build TensorFlow input pipelines	Data augmentation

Unlike tf.data.Dataset.range which will stop at some ending number, Counter will produce elements indefinitely.

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]

Args
`start`	(Optional.) The starting value for the counter. Defaults to 0.
`step`	(Optional.) The step size for the counter. Defaults to 1.
`dtype`	(Optional.) The data type for counter elements. Defaults to `tf.int64`.

Returns
A `Dataset` of scalar `dtype` elements.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/data/experimental/Counter