在 Tensorflow 中为变量赋值

Assign value to Variable in Tensorflow

我需要将零分配给 Tensorflow 变量,如下所示。有可能这样做吗?我该怎么办?

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

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

不能直接赋值,但可以使用.assign()操作实现,here is the documentation。此外,在您分配后,您必须使用 .run().eval() 运行 操作。在我看来,这段代码应该有效:

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

请找到这个以供参考。