Keras 中的 Tensorflow 自定义损失函数 - 在张量上循环

Joh*_*ohn 5 python keras tensorflow

我正在尝试在 Keras 中编写自定义损失函数,如下所示:Keras 中的自定义损失函数

我的自定义函数应该降低小于 25 的值的影响。为此,该函数将首先调用tf.clip_by_value(因此,任何小于 25 的值都将变为 25)yTrueyPred(如上面的链接所示)。接下来,为了减少这些值的影响,当一个值 inyTrue等于 25 时,将对应的值yPred改为 yPred[i] - (yPred[i] - yTrue[i])/2。最后,该函数将在这两个张量上返回“binary_crossentropy”的结果。

我有两个主要问题:

  1. 我如何实现这个自定义损失函数?我在迭代张量时遇到问题(如Looping over a tensor 中所示
  2. 我提出的函数是一个很好的实现来减少这些值的影响还是有更好的选择?

在此先感谢您的帮助