Произвольный доступ к элементу в наборе данных TF (2.0) - PullRequest
0 голосов
/ 29 апреля 2020

Предположим, у меня есть набор данных TensorFlow, и я использую TF 2.0. Я могу перебрать каждый элемент с помощью for l oop:

import tensorflow_datasets as tfds
ds = tfds.load('coco', data_dir='D:\\DataSet\\COCO')
test_data = ds["test"]
for rec in test_data:
     print(rec['image'])  

Возможно ли получить прямой доступ к N-му элементу?
что-то вроде rec = test_data[N]?

...