我有一个张量的形状(30, 116, 10)
,我想交换前两个维,以便我有一个张量的形状(116, 30, 10)
我看到numpy是实现(np.swapaxes
)的函数,我在张量流中搜索了类似的内容,但没有发现任何东西。
你有什么主意吗?
tf.transpose
提供了与相同的功能np.swapaxes
,只是形式更为通用。在你的情况下,你可以执行tf.transpose(orig_tensor, [1, 0, 2])
相当于的操作np.swapaxes(orig_np_array, 0, 1)
。
如果我不知道输入张量的尺寸,但是我确定要交换最后两个轴怎么办?就像,我应该对张量变量做些什么,以使形状的输入
(2, 3, 4, 5)
最终像一样,(2, 3, 5, 4)
但同样应该在形状的输入上起作用(3, 4, 5, 6, 7)
(并将其转换为(3, 4, 5, 7, 6)
)@KonstantinosBairaktaris看到我的答案