딥러닝/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]