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

Question

文档

我只是好奇你如何告诉它最小化哪些变量.例如,在这个线性回归代码中,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？它只是看到那些是会话中唯一的变量吗？

Answer 1

它tf.trainable_variables()包含从中包含用trainable=Trueflag 创建的所有变量(默认值)