如何使用索引更新张量的元素?

时间:2016-09-10 11:54:10

标签: python indexing tensorflow

我正在寻找一个“更新”函数,它接受一个张量t1,一些索引和值并返回一个新的张量t2,它是t1但是带有值在指数发生了相应的变化。

这似乎是最基本的功能,但我没有在文档中看到它。

我看到的是tf.scatter_update,它会更新Variable中的值:这是我想要的方式。我可以想象为Variable构建一个t2(这会工作吗?),但t2不应该是一个有状态变量,我会用任意初始值填充它。

或者,我可以手动构建一个Python嵌套列表,分别添加t1t2的每个元素,然后调用tf.convert_to_tensor。我可能错了,但这似乎非常低效。

1 个答案:

答案 0 :(得分:2)

您可以尝试tf.scatter_updatetf.scatter_add根据索引更新张量值。

请参阅问题adjust-single-value-within-tensor-tensorflow以获取参考。