张量流 - 优化变量的一部分



假设我正在优化 Ax = b,其中 A 是矩阵,x,b 是向量。我的问题 - 是否可以仅在 A 的子集上优化它?具体来说,A的补丁。换句话说,我想在 A 中保持参数子集不变。在TensorFlow中可能吗?

我想过使用 tf.silce(),但它创建了变量的新引用

谢谢!

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

您可以使用

优化器的tf.stop_gradientvar_list参数。

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

最新更新