Keras中的自定义损失函数（非对称MAE）

Question

我正在尝试为Keras LSTM实现一个自定义损失函数，它将代表非对称MAE（惩罚右移和奖励与实际相关的预测的左移）。考虑到输入参数是张量，而不是numpy数组，语法是什么？

def amae(a, p):
    product = a
    product[1:] = a[1:] - a[:-1]
    product[0] = 0
    product = -product*10
    delta = p - a
    delta = abs(delta) + (delta * product)

    return sum(delta)/len(delta)