Skip to main content
 首页 » 编程设计

python之Tensorflow GradientDescentOptimizer之它如何连接到 tf.Variables

2025年05月04日62bjzhanghao

Documentation

我很好奇您如何告诉它最小化哪些变量。例如,在这个线性回归代码中,TF 在没有被告知变量名称的情况下可以很好地优化权重/偏差:

y = W * x + b 
 
cost_func = tf.nn.l2_loss(y_ - y)  # squared error 
 
trainer = tf.train.GradientDescentOptimizer(0.01).minimize(cost_func) 

tensorflow怎么知道我要它更新W和b?它是否只是看到这些是 session 中唯一的变量?

请您参考如下方法:

它从 tf.trainable_variables() 获取它们,其中包括使用 trainable=True 标志(默认)创建的所有变量