将张量流转置转换为 pytorch

【问题标题】：converting tensorflow transpose to pytorch将张量流转置转换为 pytorch
【发布时间】：2021-11-25 03:56:40
【问题描述】：

什么是等价函数

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

在pytorch? 例如，如果我有

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

如何在pytorch 中将y 转换为x？

【问题讨论】：

【解决方案1】：

你可以使用torch.Tensor.permute:

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

【讨论】：