如何在张量流中改变三维张量的尺寸？

Question

我熟悉theano dimshuffle函数，它允许张量的维数置换。我正在寻找tensorflow中的等效函数（目前使用0.8.0）。

鉴于三维张量x，其索引为x[a,b,c]，我想这样做x.dimshuffle(1,0,2)让我能指数[b,a,c]。

它是否支持？如果不是，我如何通过使用已经实现的方法来达到相同的结果？

Answer 1

实际上，我认为tf.transpose（）正是你所需要的。只需在函数的第二个参数中将所需的排列作为列表传递即可。 详情请参阅： https://www.tensorflow.org/versions/r0.8/api_docs/python/array_ops.html#transpose