Можно ли сделать перекрестную складку с этим набором данных в тензорном потоке? от генератора

Question

train_dataset = tf.data.Dataset.from_generator(trainGenerator, (tf.float32, tf.float32), (tf.TensorShape([40,40,9]),tf.TensorShape([40,40,1])))
train_dataset = train_dataset.batch(512).prefetch(1)

Я хочу выполнить перекрестное сгибание с этим набором данных. это возможно?