딥러닝/PyTorch

[PyTorch] .permute() 차원 순서 바꾸기

Present_Kim 2021. 10. 29. 17:15
x = torch.rand(64, 32, 3)
y = x.permute(2, 0, 1)
y.size()
--->[3, 64, 32]