[iamvc3xbdd.png]
交换第 0 个维度和第 1 个维度 (转置) 为形状为 (4, 3) 的矩阵。...比如对于图片张量来说,在 PyTorch 中将通道维度放到最后面 [b, h, w, c],而在 TensorFlow 中将通道维度放在前面 [b, c, h, w],如果需要将 [b, h, w, c...换句话说,如果修改了交换维度后的张量,原始张量也会发生对应的改变;
由于 2D 张量仅有两个维度,交换维度的操作固定,类似对矩阵进行转置操作,因此 PyTorch 提供了一个更方便的方法 torch.t...比如将图片张量 [b, h, w, c] 转换为 [b, c, h, w]。...来看看如何通过 permute 函数将图片张量 [b, h, w, c] 转换为 [b, c, h, w]。