python - 如何在张量流中的张量中执行项目分配？

Question

让一个张量成为a = [0,0,0,0,0,0,0,0]另一个张量b = [1,3,0,5]，在这里我想要一个张量流操作将 1 放入张量中a，并从张量中获取位置值b。因此输出张量将是，[1,1,0,1,0,1,0,0]。如何解决这个问题呢？

标签： pythontensorflowneural-network

Answer 1

那这个呢 ？

a = tf.Variable([0,0,0,0,0,0,0,0])
b = tf.Variable([1,3,0,5])

with tf.Session() as sess:

    sess.run(tf.global_variables_initializer())

    update = tf.scatter_update(a,
                               b,
                               tf.tile(tf.constant([1],
                                       tf.int32),
                               b.shape))

    print(update.eval(session=sess))

输出是

[1 1 0 1 0 1 0 0]