即使它本身没有封闭形式,如何使用损失函数封闭形式梯度

数据挖掘 张量流
2022-02-18 20:27:10

我有一个没有封闭分析形式的损失函数,我想在 tensorflow 中实现它,它有一个很好的干净梯度(即我有一些很好的梯度函数f(x)我想用于反向传播,但是xf(x) dx没有封闭形式)

当然,我无法评估损失,但我想使用f(x)在优化过程中。这怎么可能实现?

1个回答

TensorFlow 允许使用tf.custom_gradient.

您可以编写一个装饰器,该装饰器将为 x 的特定值返回特定的梯度值。损失函数不需要评估。