tensorflow2のDatasetにもrangeが用意されています。
pythonの組み込み関数のrangeと同じようにrange(start, stop, step)というふうに使えます。
サンプルコードを見たほうがわかりやすいのでサンプルコードを見てみます。
サンプルコード
>>> import tensorflow as tf
>>> ds = tf.data.Dataset.range(1,10)
>>> list(ds.as_numpy_iterator())
[1, 2, 3, 4, 5, 6, 7, 8, 9]
>>> ds = tf.data.Dataset.range(2,5)
>>> list(ds.as_numpy_iterator())
[2, 3, 4]
>>> ds = tf.data.Dataset.range(1,10,2)
>>> list(ds.as_numpy_iterator())
[1, 3, 5, 7, 9]参考文献
https://www.tensorflow.org/api_docs/python/tf/data/Dataset#range
