[zpw858a45c.png]
前言
交换维度顾名思义就是交换不同的维度,线性代数中矩阵的转置操作可以看成是交换第 0 个和第 1 个维度。比如下图形状为 (3, 4) 的矩阵。...[iamvc3xbdd.png]
交换第 0 个维度和第 1 个维度 (转置) 为形状为 (4, 3) 的矩阵。...比如对于图片张量来说,在 PyTorch 中将通道维度放到最后面 [b, h, w, c],而在 TensorFlow 中将通道维度放在前面 [b, c, h, w],如果需要将 [b, h, w, c...交换维度的操作至少要求张量拥有两个以及两个以上的维度才有意义,因此在介绍交换维度的方式时不再考虑 0D 和 1D 张量。...下面是在使用 transpose 函数时的几个注意事项。