如何通过一个或多个索引更新4D Tensorflow变量中的值？

Question

假设我有一个4D张量，并且我想更改其中的一个元素：

[[[[0, 0, 0], [0, 0, 0]], [[0, 0, 0], [0, 0, 0]]]]

到

[[[[0, 0, 0], [0, 0, 0]], [[0, 1, 0], [0, 0, 0]]]]

任何想法都可以轻松实现吗？我知道tf.scatter_update，但似乎性能相当庞大，因为在进行更新时需要更改整个尺寸。

Answer 1

事实证明，tf.scatter_nd_update是用于更新单个值的操作。另一方面，tf.scatter_update将仅更新行/切片。