Numpy array[indices] =标量的流等效

Question

我有一个张量和一个同等级的指数张量。我想把与指数张量中的指数对应的张量的值设置为某个标量。我该怎么做？

换句话说，我正在寻找与以下Numpy操作等价的Tensorflow：

array[indices] = scalar

在我的具体例子中，我们讨论的是一维张量：

mask = tf.zeros_like(some_1D_tensor)

(e.g. mask = [0, 0, 0, 0, 0])

设indices是一个一维张量，它包含我想要设置为标量值1的mask索引。

mask[indices] = 1

(e.g. for indices = [1, 3] the output should be mask == [0, 1, 0, 1, 0])

Answer 1

我不知道它以前是不是在那里，或者我只是没见过，但是一般的情况相当于

array[indices] = scalar

是

tensor = tf.scatter_nd_update(tensor, indices, updates)

使用更新()