message: [tf.data] Update documentation for tf.data.experimental.Counter
diff: @@ -29,15 +29,26 @@ from tensorflow.python.util.tf_export import tf_export
  def CounterV2(start=0, step=1, dtype=dtypes.int64):
  """Creates a `Dataset` that counts from `start` in steps of size `step`.
  - For example:
  + Unlike `tf.data.Dataset.range` which will stop at some ending number,
  + `Counter` will produce elements indefinitely.
  - ```python
  - 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, ...)
  - ```
  + >>> 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.
  