tf.data.experimental.Counter
Creates a Dataset that counts from start in steps of size step.
tf.data.experimental.Counter(
start=0, step=1, dtype=tf.dtypes.int64
)
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, ...)
| 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. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/data/experimental/Counter