Tensorflow - 优化变量的一部分

时间:2017-10-17 09:41:35

标签: optimization tensorflow

让我们说我优化Ax = b,其中A是矩阵,x,b是向量。 我的问题 - 是否可以仅在A的子集上进行优化?特别是A的补丁 换句话说,我想保持A中参数的一个子集不变。 TensorFlow有可能吗?

我考虑过使用tf.silce(),但它会创建一个新的变量引用

谢谢!

2 个答案:

答案 0 :(得分:0)

除非我误解了您的问题(或者缺少上下文),否则只需使用tf.Variable()定义要优化的A部分,并定义您不需要的部分。使用tf.Constant()。

答案 1 :(得分:0)

您可以使用优化程序的tf.stop_gradientvar_list参数。

有关详细信息,请参阅此答案:https://stackoverflow.com/a/34478044/4554460