tf.data.experimental.at

Returns the element at a specific index in a datasest.

Currently, random access is supported for the following tf.data operations:

dataset A tf.data.Dataset to determine whether it supports random access.
index The index at which to fetch the element.

A (nested) structure of values matching tf.data.Dataset.element_spec.
Raises UnimplementedError: If random access is not yet supported for a dataset.