我想修改以下keras均方误差损失（MSE），以便仅稀疏计算损失。
def mean_squared_error(y_true, y_pred):
return K.mean(K.square(y_pred - y_true), axis=-1)
我的输出y是3通道图像，其中第3通道仅在要计算损耗的那些像素处不为零。知道如何修改以上内容以计算稀疏损失吗？
machine-learning computer-vision neural-network conv-neural-network keras