Keras:如何在每批训练时修改keras模型的输入

from keras.utils import Sequence

class CustomDataGenerator(Sequence):
    def __init__(self, x, y, batch_size):
        self.x = x
        self.y = y
        self.batch_size = batch_size

    def __len__(self):
        return len(self.x) // self.batch_size

    def __getitem__(self, idx):
        batch_x = self.x[idx * self.batch_size:(idx + 1) * self.batch_size]
        batch_y = self.y[idx * self.batch_size:(idx + 1) * self.batch_size]

        # 在这里修改输入数据
        modified_batch_x = modify_input(batch_x)

        return modified_batch_x, batch_y

# 创建自定义数据生成器
data_generator = CustomDataGenerator(x_train, y_train, batch_size)

# 使用自定义数据生成器训练模型
model.fit_generator(data_generator, epochs=10)