B ٻd @sdZddlmZddlmZddlmZddlmZddlm Z ddl m Z e dgd e d d dd ej fd dZe dgd e d d dd ej fddZeje_ereZneZd S)zThe Counter Dataset.)tf2) counter_op) dataset_ops)dtypes) deprecation) tf_exportzdata.experimental.Counter)Zv1Nz+Use `tf.data.Dataset.counter(...)` instead.cCst|||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) >>> 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. )rcounter)startstepdtyper a/var/www/html/venv/lib/python3.7/site-packages/tensorflow/python/data/experimental/ops/counter.py CounterV2s#rcCstt|||S)N)rZDatasetV1Adapterr)r r r r r r CounterV1>sr)__doc__Ztensorflow.pythonrZtensorflow.python.data.opsrrZtensorflow.python.frameworkrZtensorflow.python.utilrZ tensorflow.python.util.tf_exportr deprecatedZint64rrenabledCounterr r r rs         $