Как заменить tf.train.batch, поскольку он устарел

Question

Это код для обучения данных mnist с помощью Petastorm.

def train_and_test(dataset_url, training_iterations, batch_size, evaluation_interval):

    with make_reader(os.path.join(dataset_url, 'train'), num_epochs=None) as train_reader:
        with make_reader(os.path.join(dataset_url, 'test'), num_epochs=None) as test_reader:
            train_readout = tf_tensors(train_reader)
            train_image = tf.cast(tf.reshape(train_readout.image, [784]), tf.float32)
            train_label = train_readout.digit
            batch_image, batch_label = tf.train.batch(
                [train_image, train_label], batch_size=batch_size
            )

Я не знаю как заменить tf.train.batch. Не могли бы вы помочь с этим.

Answer 1

Вы можете использовать dataset.batch с участием tf.data.Dataset и petastorm также поддерживает tf.data.Dataset который упоминается на их веб-сайте.

Для кода по реализации tf.data.Dataset с участием petastormВы можете получить это здесь.
Подробнее о dataset.batchВы можете найти это здесь.