首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >GradientTape不计算梯度

GradientTape不计算梯度
EN

Stack Overflow用户
提问于 2020-06-01 10:43:41
回答 1查看 283关注 0票数 0

我理解,只要我在tf.GradientTape()上下文中定义一个计算,梯度磁带就会计算出计算输出所依赖的所有变量。但是,我认为我并没有完全理解梯度的子属性,因为下面的代码没有像我期望的那样执行:

代码语言:javascript
运行
复制
import tensorflow as tf
x = tf.Variable(2.)
loss_ = x**2-2*x+1
with tf.GradientTape(persistent=True) as g:
    loss = loss_*1
print(g.gradient(loss,x))
output: None

为什么不计算梯度wrt?

我只能计算wrt到上下文中显式使用的变量的梯度。例如,以下代码也不计算梯度:

代码语言:javascript
运行
复制
import tensorflow as tf
tf.compat.v1.disable_eager_execution()
x = tf.Variable(2.)
t1 = x**2
t2 = -2*x
t3 = 1.
with tf.GradientTape(persistent=True) as g:
    loss = t1+t2+t3
print(g.gradient(loss,x))
EN

回答 1

Stack Overflow用户

发布于 2020-06-01 11:06:48

GradientTape对象gwith语句结束后超出作用域。

换句话说,尝试在with语句中打印渐变。

下面是对我有用的东西:

代码语言:javascript
运行
复制
def get_gradients(inputs, target, model):
    with tf.GradientTape() as tape:
        prediction = model(inputs)
        loss = loss_object(target, prediction)
        gradient = tape.gradient(loss, model.trainable_variables)

    return gradient
票数 1
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/62130393

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档