[iamvc3xbdd.png]
交换第 0 个维度和第 1 个维度 (转置) 为形状为 (4, 3) 的矩阵。...换句话说,如果修改了交换维度后的张量,原始张量也会发生对应的改变;
由于 2D 张量仅有两个维度,交换维度的操作固定,类似对矩阵进行转置操作,因此 PyTorch 提供了一个更方便的方法 torch.t...比如将图片张量 [b, h, w, c] 转换为 [b, c, h, w]。...(imgs.shape)
torch.Size([1, 3, 32, 28])
虽然能够将图片张量 [b, h, w, c] 转换为 [b, c, h, w],但是使用了两次 transpose 函数...来看看如何通过 permute 函数将图片张量 [b, h, w, c] 转换为 [b, c, h, w]。