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

时间:2016-06-17 21:08:47

标签: python tensorflow

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如何知道我希望它更新Wb?它只是看到那些是会话中唯一的变量吗?

1 个答案:

答案 0 :(得分:7)

它来自tf.trainable_variables(),包括用trainable=True标志创建的所有变量(默认值)