如何使用Keras實現自定義丟失函數？

Question

有一張紙Beyond MSE

在本文中，筆者提出了一種梯度差損失函數。 gradient difference loss function，結合MSE，預測結果會較不模糊。

我是凱拉斯的新手，你能告訴我一個方向怎麼實現嗎？

Answer 1

使用keras後端。 如果後端是張量流，則使用K.conv3d來計算梯度。