如何冻结张量流中变量的某些部分

Question

我注意到可以通过设置trainable=False冻结变量。我想知道如何冻结其中的一部分，而不是将其分为两个变量。

例如，我只想冻结v = tf.Variable(tf.zeros([2])中的第一个元素。看来grad_new = tf.multiply(grad, np.array([0,1]))无效。

Answer 1

问题解决了。

grad_new = tf.multiply(grad, np.array([0,1]))实际上效果很好。是使代码混乱的保护程序。