tensorflow - 我可以在 2.0 计算 NN 输出 wrt 输入的 autograd 中重新创建 tf1.0 代码吗？

Question

我不明白如何通过删除 tf2 中的图形节点定义来获得这样的渐变对象。我怎样才能在训练期间仍然指示 tf.gradients() 给我这个 autodiff 对象。

#x,y,t,u = tf.placeholder(...)

u = self.neural_net(tf.concat([x,y,t], 1), self.weights, self.biases)    
du_dt = tf.gradients(psi, t)[0]

标签： tensorflow