如何在张量流中改组 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
我熟悉允许张量维度排列的 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