📌  相关文章
📜  pytorch 张量更改维度顺序 - Python 代码示例

📅  最后修改于: 2022-03-11 14:45:21.336000             🧑  作者: Mango

代码示例1
a = torch.rand(1,2,3,4)
print(a.transpose(0,3).transpose(1,2).size())
print(a.permute(3,2,1,0).size())