在Tensorflow中为Variable赋值

Question

我需要为Tensorflow变量赋值为零，如下所示。有可能吗？我该怎么办？

layer = tf.nn.sigmoid(tf.add(tf.matmul(x, h), b))
layer[0] = 0

在上面的例子中，我想将零值设置为通过乘以x矩阵和h向量计算的变量的第一个索引。

Answer 1

您无法直接分配，但可以使用.assign()操作来实现，here is the documentation。分配后，您还必须使用.run()或.eval()运行操作。在我看来，这段代码应该有效：

layerOp = layer[0].assign(0)
sess.run(layerOp) 

请参阅此post以供参考。