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