我有一个形状为 (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)
.
关于tensorflow - 如何在 TensorFlow 中交换张量的 Axis ?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/38212205/