小编Pie*_*rre的帖子

在给定两个索引的情况下,在张量中交换两个值的最佳方法是什么？

这是我到目前为止使用变量所做的.

import numpy as np
import tensorflow as tf

indices = tf.constant([0,1])
data = tf.Variable([1,2,8])
data = tf.scatter_update(data,indices,tf.gather(data,tf.reverse(indices,[0])))
init_op = tf.initialize_all_variables()

sess = tf.Session()
sess.run([init_op])
print "data:", sess.run(data)

它给了我

a:[2 1 8]

我确定这很难看,还有另外一种方法吗？

问候,