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)