Tensorflow：尝试通过张量值访问变量的索引？

Question

我正在尝试模仿 TensorFlow 中 numpy 的以下行为。

z = np.zeros(2 * 10 - 1, dtype=np.float32)
z[[2,10]] = 1

我有什么

test = tf.Variable(tf.zeros(2 * 10 - 1, dtype=tf.float32))
test[tf.constant([2,10])].assign(1)

我需要变量，因此不能只使用常量零。

尝试此操作时出现错误

InvalidArgumentError: Shape must be rank 1 but is rank 2 for 'strided_slice' (op: 'StridedSlice') with input shapes: [19], [1,2], [1,2], [1].

但这没有意义，因为我提供的索引是 1 级的，并且由于某种原因它被重新调整。

我将如何模仿上述行为？

Answer 1

这似乎最接近我所寻找的，但我担心这会产生重复值，如果测试很大，那么这将是巨大的。

tf.scatter_add(test,[2,10],1)

欢迎提供更好的答案。