[zpw858a45c.png]
前言
交换维度顾名思义就是交换不同的维度,线性代数中矩阵的转置操作可以看成是交换第 0 个和第 1 个维度。比如下图形状为 (3, 4) 的矩阵。...比如对于图片张量来说,在 PyTorch 中将通道维度放到最后面 [b, h, w, c],而在 TensorFlow 中将通道维度放在前面 [b, c, h, w],如果需要将 [b, h, w, c...比如将图片张量 [b, h, w, c] 转换为 [b, c, h, w]。...([1, 3, 32, 28])
虽然能够将图片张量 [b, h, w, c] 转换为 [b, c, h, w],但是使用了两次 transpose 函数,并且需要熟知每次变换后对应维度的位置,非常容易出错...来看看如何通过 permute 函数将图片张量 [b, h, w, c] 转换为 [b, c, h, w]。