Keras Пользовательская функция потерь для передачи аргумента (массива с нулевым значением) с тем же размером пакета, что и для y_true и y_pred

Question

Я реализовал пользовательскую функцию потерь, которая принимает дополнительный шум (массив numpy), как показано ниже:

  def custom_rcae_loss(self):

        N = self.Noise
        lambda_val = self.lamda[0]
        mue = self.mue
        self.batchNo += 1
        index = self.batchNo
        def custom_rcae(y_true, y_pred):
            if(N.ndim >1):
                term1 = keras.losses.mean_squared_error(y_true, (y_pred + N ))

Проблема в том, что y_pred имеет форму (batch_size, 28,28,1):Как я могу убедиться, что мой шум также имеет ту же форму, что и y_pred?Так как я хотел бы выступить (y_pred + Noise).

Например: если мой ввод 5983, то количество сэмплов с размером партии 128 Не существует такого же количества разбиений batch_size.

Как мы можем решить эту проблему, используя keras, чтобы удостовериться, что шум имеет ту же форму, что и y_pred

Ждем предложений и подсказок Заранее спасибо