如何使用tensorflow正确交换数组中的两个值?

时间:2017-05-02 21:16:08

标签: tensorflow

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

以下是我目前使用变量做的事情。

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]

我确定这很难看,还有其他办法吗?

此致

0 个答案:

没有答案