tensorflow2.0 自动求导和自定义训练

tensorflow2.0-自动求导 1 tf.GradientTape() tf.GradientTape()是常用的自动求导方法,分为三种情况,一阶导数,高阶导数,和多次求导。 一阶导数: 如果是常数(tf.constant),需要调用watch()方法进行手动监视,如果是变量则不需要,在model如果变量的trainable=True则自动监视。 x = tf.constant(3.0) with tf.GradientTape() as tape: # 需要调用 g.watch(x) y = x * x dy_dx = g.gradient(y, x) x = tf.Var