tf.contrib.data.get_single_element

Возвращает единственный элемент в dataset в виде вложенной структуры тензоров. (устарело)

tf.contrib.data.get_single_element(
    dataset
)

Эта функция позволяет использовать tf.data.Dataset в бессостоятельном выражении «тензор на вход — тензор на выход», не создавая tf.compat.v1.data.Iterator. Это может быть полезно, когда ваши преобразования предобработки выражаются как Dataset, и вы хотите использовать преобразование во время предоставления. Например:

input_batch = tf.compat.v1.placeholder(tf.string, shape=[BATCH_SIZE])

def preprocessing_fn(input_str):
  # ...
  return image, label

dataset = (tf.data.Dataset.from_tensor_slices(input_batch)
           .map(preprocessing_fn, num_parallel_calls=BATCH_SIZE)
           .batch(BATCH_SIZE))

image_batch, label_batch = tf.data.experimental.get_single_element(dataset)