在给定两个索引的情况下,在张量中交换两个值的最佳方法是什么?
以下是我目前使用变量做的事情。
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]
我确定这很难看,还有其他办法吗?
此致