При проверке кода потери кераса я обнаружил, что среднеквадратическая ошибка потери определяется следующим образом:
def mean_squared_error(y_true, y_pred):
return K.mean(K.square(y_pred - y_true), axis=-1)
Но, когда y_true является 4-мерным тензором с формой (n, h,w, c) эта функция, кажется, возвращает результат трехмерного тензора, и она должна вызвать ошибку.
Однако при обучении эта функция по-прежнему работает нормально.Что произошло, когда я тренировал нейронную сеть при использовании этой функции.
Спасибо за ваш ответ