Keras 自定义损失：如何更新 y_true 和 y_prediction 输入？ （值错误）

Question

我想为时间序列数据的 RNN (GRU) 模型构建自定义损失函数。

y_true 和 y_prediction 是来自完整时间序列数据的长度为 N 的片段。

我需要确保每次数据的斜率/轨迹向上，并且如果值下降，则仅使用下降之前的数据（以缩短时间序列）。 功能代码为：

def custom_mae_loss(y_true,y_pred):
  # check y_true doesnt have drops
  for i in range(len(y_true)-1):  
    if y_true[i+1] < y_true[i]:
      idx_to_stop = i
      y_true = y_true[:idx_to_stop]
      y_pred = y_pred[:idx_to_stop]
      break

  total_err = 0
  for yt,yp in zip(y_true, y_pred):
    total_err += abs(yt-yp)
  mae = total_err / len(y_true)

但是，当我使用此功能运行网络时，出现以下错误：

  ValueError: condition of if statement expected to be `tf.bool` scalar, got Tensor("custom_mae_loss/while/Less:0", shape=(40, 1), dtype=bool); to check for None, use `is not None`

知道如何解决这个问题吗？