我有一个形状的张量（30,116,10），我想交换前两个维度，所以我有一个张量的形状（116,30,10）
我看到numpy实现了这样一个函数（np.swapaxes），我在tensorflow中搜索了类似的东西，但我什么都没发现 .
你有什么主意吗 ？
提前致谢，
亚历克西斯
tf.transpose提供与 np.swapaxes 相同的功能，但采用更通用的形式 . 在你的情况下，你可以做 tf.transpose(orig_tensor, [1, 0, 2]) ，这相当于 np.swapaxes(orig_np_array, 0, 1) .
np.swapaxes
tf.transpose(orig_tensor, [1, 0, 2])
np.swapaxes(orig_np_array, 0, 1)
1 回答
tf.transpose提供与
np.swapaxes相同的功能，但采用更通用的形式 . 在你的情况下，你可以做tf.transpose(orig_tensor, [1, 0, 2])，这相当于np.swapaxes(orig_np_array, 0, 1).