tf.data.experimental.enumerate_dataset | TensorFlow v2.16.1 (original) (raw)

tf.data.experimental.enumerate_dataset

A transformation that enumerates the elements of a dataset. (deprecated)

View aliases

Compat aliases for migration

tf.compat.v1.data.experimental.enumerate_dataset

tf.data.experimental.enumerate_dataset(
    start=0
)

It is similar to python's enumerate. For example:

# NOTE: The following examples use `{ ... }` to represent the
# contents of a dataset.
a = { 1, 2, 3 }
b = { (7, 8), (9, 10) }

# The nested structure of the `datasets` argument determines the
# structure of elements in the resulting dataset.
a.apply(tf.data.experimental.enumerate_dataset(start=5))
=> { (5, 1), (6, 2), (7, 3) }
b.apply(tf.data.experimental.enumerate_dataset())
=> { (0, (7, 8)), (1, (9, 10)) }

Args
start	A tf.int64 scalar tf.Tensor, representing the start value for enumeration.

Returns
A Dataset transformation function, which can be passed totf.data.Dataset.apply.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.

Last updated 2024-04-26 UTC.