2016-06-17 2 views
1

DocumentationTensorflow GradientDescentOptimizer - как он подключается к tf.Variables?

Мне просто интересно, как вы говорите, чтобы минимизировать переменные. Например, в этой линейной регрессии коде, 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? Видно ли, что это единственные переменные в сеансе?

ответ

5

Он принимает их от tf.trainable_variables(), который включает в себя все переменные, созданные с помощью флага trainable=True (по умолчанию)

Смежные вопросы