Returns the element at a specific index in a datasest.
tf.data.experimental.at(
dataset, index
)
Currently, random access is supported for the following tf.data operations:
tf.data.Dataset.from_tensor_slices,tf.data.Dataset.from_tensors,tf.data.Dataset.shuffle,tf.data.Dataset.batch,tf.data.Dataset.shard,tf.data.Dataset.map,tf.data.Dataset.range,tf.data.Dataset.zip,tf.data.Dataset.skip,tf.data.Dataset.repeat,tf.data.Dataset.list_files,tf.data.Dataset.SSTableDataset,tf.data.Dataset.concatenate,tf.data.Dataset.enumerate,tf.data.Dataset.parallel_map,tf.data.Dataset.prefetch,tf.data.Dataset.take,tf.data.Dataset.cache (in-memory only)
Users can use the cache operation to enable random access for any dataset, even one comprised of transformations which are not on this list. E.g., to get the third element of a TFDS dataset:
ds = tfds.load("mnist", split="train").cache()
elem = tf.data.Dataset.experimental.at(ds, 3)
| Args | |
|---|---|
dataset | A tf.data.Dataset to determine whether it supports random access. |
index | The index at which to fetch the element. |
| Returns | |
|---|---|
A (nested) structure of values matching tf.data.Dataset.element_spec. | |
Raises | UnimplementedError: If random access is not yet supported for a dataset. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/data/experimental/at