tf.data.experimental.at

Возвращает элемент по заданному индексу в наборе данных.

tf.data.experimental.at(
    dataset, index
)

В настоящее время произвольный доступ поддерживается для следующих операций tf.data:

• 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 (только в оперативной памяти)

  Пользователи могут использовать операцию кеширования для включения произвольного доступа к любому набору данных, даже если он состоит из преобразований, которые не входят в этот список. Например, чтобы получить третий элемент набора данных TFDS:

  ds = tfds.load("mnist", split="train").cache()
  elem = tf.data.Dataset.experimental.at(ds, 3)