BVh dZddlmZddlmZddlmZddlmZddl m Z ddl m Z e dg e jd d dd ejfd Ze dg e jd d dd ejfdZeje_ej"reaneadZej(ey )zThe Counter Dataset.)tf2) v2_compat) dataset_ops)dtypes) deprecation) tf_exportzdata.experimental.Counter)v1Nz+Use `tf.data.Dataset.counter(...)` instead.cDtjj|||S)aCreates a `Dataset` that counts from `start` in steps of size `step`. Unlike `tf.data.Dataset.range` which will stop at some ending number, `Counter` will produce elements indefinitely. >>> dataset = tf.data.experimental.Counter().take(5) >>> [a.item() for a in 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) >>> [a.item() for a in dataset.as_numpy_iterator()] [2, 3, 4, 5, 6] >>> dataset = tf.data.experimental.Counter(start=2, step=5).take(5) >>> [a.item() for a in dataset.as_numpy_iterator()] [2, 7, 12, 17, 22] >>> dataset = tf.data.experimental.Counter(start=10, step=-1).take(5) >>> [a.item() for a in 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. )rDatasetcounterstartstepdtypes _/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/data/experimental/ops/counter.py CounterV2rs F    $ $UD% 88cBtjt|||SN)rDatasetV1Adapterrrs r CounterV1r>s  % %itU&C DDrcFtjrtaytayr)renabledrCounterrrr _tf2_callbackrLs[[]GGr)__doc__tensorflow.pythonrtensorflow.python.compatrtensorflow.python.data.opsrtensorflow.python.frameworkrtensorflow.python.utilr tensorflow.python.util.tf_exportr deprecatedint64rrrrrregister_data_v2_callbackrrrr(s!.2..6 &2.KLAV\\!9M/!9H *+,KLAV\\EM-E%% 3;;= ' '$ ##M2r