torch - PyTorch 中的无量纲转置

Question

PyTorch 的 torch.transpose函数仅转置 2D 输入。文档是 here .

另一方面，Tensorflow 的 tf.transpose函数允许您转置 N 的张量任意维度。

有人可以解释为什么 PyTorch 没有/不能具有 N 维转置功能吗？这是由于 PyTorch 与 Tensorflow 的定义然后运行范式中计算图构建的动态性质吗？

Answer 1

它在 pytorch 中只是叫法不同。 torch.Tensor.permute将允许您在 pytorch 中交换维度，就像 tf.transpose 在 TensorFlow 中所做的那样。

作为如何将 4D 图像张量从 NHWC 转换为 NCHW 的示例(未测试，因此可能包含错误):

>>> img_nhwc = torch.randn(10, 480, 640, 3)
>>> img_nhwc.size()
torch.Size([10, 480, 640, 3])
>>> img_nchw = img_nhwc.permute(0, 3, 1, 2)
>>> img_nchw.size()
torch.Size([10, 3, 480, 640])