我很好奇您如何告诉它最小化哪些变量。例如,在这个线性回归代码中,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 标志(默认)创建的所有变量
