python - Keras/Tensorflow：通道顺序优先和损失函数轴 -1

Question

我从Keras 源代码中看到，例如，损失函数mae定义为

def mean_absolute_error(y_true, y_pred):
    return K.mean(K.abs(y_pred - y_true), axis=-1)

假设通道位于末尾，这将产生一个对每个像素NHW都有损失的张量。mae

从这个问题Why use axis=-1 in Keras metrics function? ，我明白在打印要记录的内容之前调用了 K.mean。

我对我的图像使用channel_first排序，即NCHW那里没有channel_order检查条件语句。

这会影响损失反向传播的方式（或者还有什么我需要担心的）？或者我应该从那里手动创建损失函数axis=1？或者，如果根本不包含轴，使损失函数返回单个标量而不是张量，是否可以？这样做的正确方法是什么？

标签： pythontensorflowkeras