如何在张量流中改组 3D 张量尺寸?

How can the 3D-tensor dimensions be shuffled in tensorflow?

我熟悉允许张量维度排列的 theano dimshuffle 函数。我正在寻找 tensorflow 中的等效函数(当前使用 0.8.0)。

给定一个三维张量 x,索引为 x[a,b,c],我想做类似 x.dimshuffle(1,0,2) 的事情,这样我就可以索引为 [b,a,c].

是否支持?如果没有,我如何使用已经实现的方法获得相同的结果?

事实上,我认为 tf.transpose() 正是您所需要的。只需将所需的排列作为函数的第二个参数中的列表传递即可。 详情请见: https://www.tensorflow.org/versions/r0.8/api_docs/python/array_ops.html#transpose