我想取损失函数的梯度,仅相对于层中的单个权重。对于关于整个第一层的导数,下面的代码很好用 with tf.GradientTape() as tape:g = tape.gradient(loss(y, y_pred), model.trainable_variables[0][:1]) 但是这将返回</e
x = tf.convert_to_tensor(np.linspace(-5,5,11))with tf.GradientTape() as g:x=tf.Variable(-5.0)with tf.GradientTape(persistent=True) as g: y.assign(x*x)
dy_dx = g.gradient(y,