有关keras.losses.mean_square_error

Question

在检查keras损失的代码时，我发现均方误差的损失定义如下：

def mean_squared_error(y_true, y_pred):
    return K.mean(K.square(y_pred - y_true), axis=-1)

但是，当y_true是形状为（n，h，w，c）的4D张量时，此函数似乎返回3D张量结果，并且应该引发错误。

但是，在训练时，该功能仍然可以正常工作。使用此功能训练神经网络时会发生什么情况。

感谢您的回答