tf.data.experimental.Counter

TensorFlow 1 version View source on GitHub

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

Used in the notebooks

Used in the guide

For example:

Dataset.count() == [0, 1, 2, ...)
Dataset.count(2) == [2, 3, ...)
Dataset.count(2, 5) == [2, 7, 12, ...)
Dataset.count(0, -1) == [0, -1, -2, ...)
Dataset.count(10, -1) == [10, 9, ...)

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.

A Dataset of scalar dtype elements.