如何在tensorflow中设置一些张量元素？

Question

我有两个张量。

Logits：

tf.Tensor 'MemN2N_1/MatMul_3:0' shape=(?, 18748) dtype=float32

第二张量包含我想要将第一张量的索引元素设置为零的索引。

案例：

tf.Tensor 'Reshape_1:0' shape=(?, 2076) dtype=int32

例如，如果我有：

logits = [[0.1, 0.2, 0.3, 0.4,...], [0.1, 0.2, 0.3, 0.4, ...], ...]

stories = [[0, 2], [1, 3], ...],

然后我想将logits张量修改为：

logits = [[0, 0.2, 0, 0.4, ...], [0.1, 0, 0.3, 0, ...], ...]

我怎样才能做到这一点？