我正在寻找一个“更新”函数,它接受一个张量t1
,一些索引和值并返回一个新的张量t2
,它是t1
但是带有值在指数发生了相应的变化。
这似乎是最基本的功能,但我没有在文档中看到它。
我看到的是tf.scatter_update
,它会更新Variable
中的值:这是我想要的方式。我可以想象为Variable
构建一个t2
(这会工作吗?),但t2
不应该是一个有状态变量,我会用任意初始值填充它。
或者,我可以手动构建一个Python嵌套列表,分别添加t1
和t2
的每个元素,然后调用tf.convert_to_tensor
。我可能错了,但这似乎非常低效。
答案 0 :(得分:2)
您可以尝试tf.scatter_update或tf.scatter_add
根据索引更新张量值。
请参阅问题adjust-single-value-within-tensor-tensorflow以获取参考。