将tensorflow转置转换为pytorch

Question

的等效函数是什么?

x = tf.transpose(y, perm=[2, 0, 1])

在pytorch?例如，如果我有

import torch
y=torch.randn(10, 480, 640)

如何将pytorch中的y转换为x?

Answer 1

您可以使用torch.Tensor.permute:

>>> x = y.permute(2, 0, 1)