Tensorflow自动区分差异

时间:2018-10-18 15:58:27

标签: python tensorflow

我正在使用tensorflow 1.11在Python中使用以下代码来计算一些导数:

c = self.y_predicted
x = self.x_tf
t = self.t_tf

c_t = tf.gradients(c, t)[0]
c_x = tf.gradients(c, x)[0]
c_xx = tf.gradients(c_x, x)[0]

这非常好用,我得到正确的结果并且收敛很快。但是,我希望将其替换为:

self.X_tf = tf.concat([self.x_tf, self.t_tf], 1)

dc = tf.gradients(self.y_predicted, self.X_tf)[0]
d2c = tf.gradients(dc, self.X_tf)[0]

c_t = dc[:,1:2]
c_xx = d2c[:,0:1]

我做了一些实验,这应该可以工作,但是收敛速度较慢,收敛成本较高。我期望得到相同的结果和收敛速度,那么我还缺少什么?

谢谢

0 个答案:

没有答案