计算外部损失函数，但计算张量流中的梯度？

Question

我想训练一个只能从外部计算损失函数的模型。因此，我获取了最后一层的输出，在外部计算了一些值，并希望使用它来更新我的网络。我可以在tensorflow中实现这样的设置吗？

Answer 1

是的，您必须在图中定义最小化的损失，例如：

loss = Network.loss(input_tensor)
loss_2 = out_function(loss)

然后只需指定：

train_step = optimizer.minimize(loss_2)

应该可以