Warm tip: This article is reproduced from serverfault.com, please click

axis-如何在TensorFlow中交换张量轴?

(axis - How do I swap tensor's axes in TensorFlow?)

发布于 2016-07-05 20:31:08

我有一个张量的形状(30, 116, 10),我想交换前两个维,以便我有一个张量的形状(116, 30, 10)

我看到numpy是实现(np.swapaxes的函数,我在张量流中搜索了类似的内容,但没有发现任何东西。

你有什么主意吗?

Questioner
Alexis Rosuel
Viewed
11
547 2018-10-06 20:06:40

tf.transpose提供了与相同的功能np.swapaxes,只是形式更为通用。在你的情况下,你可以执行tf.transpose(orig_tensor, [1, 0, 2])相当于的操作np.swapaxes(orig_np_array, 0, 1)