标签: tensorflow keras loss-function
我正在尝试构建自定义损失函数,该函数利用模型预测的输出的梯度(即y_pred的梯度)。例如,损失函数可能试图使模型输出的梯度(grad(y_pred))和y_true之间的均方误差最小。
尽管可以在自定义损失函数中进行y_pred平方等许多操作,但在自定义损失函数中不可能使用其梯度。我知道这可能是因为y_pred只是对计算图的引用,并且最初没有任何值。有人可以建议解决方法吗?