Python keras中的自定义损失函数返回nan损失

以下是我在keras中的自定义损失函数。

y\u true

和

y\u pred

都具有形状

（256384,3）

---

在这里，我将一部分差异切分，并找到该差异的

均方损失
。但是，此函数将返回
nan
作为丢失。为什么会这样？
您确定
y\u pred
没有
nan
？有时，模型梯度可能是
nan
，把一切都搞砸了…@Richard\w当我使用标准的keras
均方误差
作为损失时，它不会给我
nan
s。因此，我认为问题不在于
y_pred
你的代码对我来说似乎是合法的…你会
diff=diff[0:256，0:384]
也给出
nan
？
def myloss(y_true,y_pred):
    diff = y_pred - y_true
    diff = diff[96:160 , 144:240]
    return K.mean(K.square(diff), axis=-1)