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

Question

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 中唯一的变量？

Answer 1

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