比如对于图片张量来说,在 PyTorch 中将通道维度放到最后面 [b, h, w, c],而在 TensorFlow 中将通道维度放在前面 [b, c, h, w],如果需要将 [b, h, w, c...transpose
torch.transpose(input, dim0, dim1) 函数将输入张量 input 的第 dim0 个维度和第 dim1 个维度进行交换,并将交换维度后的张量返回。...比如将图片张量 [b, h, w, c] 转换为 [b, c, h, w]。...permute
前面提到过 PyTorch 从接口的角度将张量的操作分成两种方式。...来看看如何通过 permute 函数将图片张量 [b, h, w, c] 转换为 [b, c, h, w]。