使用Tensorflow进行Numpy索引操作

Question

我想更新张量的值，如：

ldream[w,:,x,y,z] = 0

但不断收到错误说：

TypeError: 'Tensor' object does not support item assignment

Answer 1

似乎ldream是一个tensorflow变量，遗憾的是它不能像numpy数组那样直接赋值。

要更新张量流中的张量值，可以进行赋值操作，然后运行（eval）该操作。 以下是如何执行此操作的示例：

Tensorflow：如何修改张量值

这样做的原因是当你编写你的tf变量和操作时，你实际上是“暂存”它们以后发生（例如当你做sess.run ）; 就像tensorflow实际运行的蓝图一样。

Answer 2

通过使用numpy数组创建布尔掩码来解决此问题：

ldream_mask = np.zeros(ldream.shape, dtype=np.bool)

然后选择所需的索引并将其标记为True

ldream_mask[w,:,x,y,z] =  True

然后使用tf.where使用ldream_mask更新所需的索引