sunshine9404的博客

原创 tensorflow中tf.gradients()解析

tf.gradients()解析及grad_ys在xs为(?, 1)时的理解 问题简介 使用tensorflow1.15学习时，有一项tf.gradients的代码，其中用到了grad_ys这个参数，经过一些解析，得到了一些自己的理解 原代码 def fwd_gradients_1(self, U, x): g = tf.gradients(U, x, grad_ys=self.dummy_x1_tf)[0] return tf.gradients(g, self.d