我有一个张量的形状(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)。
本文收集自互联网，转载请注明来源。
如有侵权，请联系[email protected] 删除。
我来说两句