tf.contrib.data.batch_and_drop_remainder

Преобразование для объединения элементов в батчи, отбрасывая последний небольшой батч (если он есть). (устаревшее)

tf.contrib.data.batch_and_drop_remainder(
    batch_size
)

Подобно tf.data.Dataset.batch, это преобразование объединяет последовательные элементы этого набора данных в батчи. Однако, если размер батча не делит размер входного набора данных нацело, это преобразование отбросит последний меньший элемент.

Следующий пример иллюстрирует разницу между этим преобразованием и Dataset.batch():

dataset = tf.data.Dataset.range(200)
batched =
dataset.apply(tf.contrib.data.batch_and_drop_remainder(128))
print(batched.output_shapes)  # ==> "(128,)" (the batch dimension is known)

В противоположность этому, dataset.batch(128) вернёт набор данных из двух элементов с форматами (128,) и (72,), поэтому размерность батча не будет статически известна.