Deep CV
[PyTorch] .permute() 차원 순서 바꾸기 본문
x = torch.rand(64, 32, 3)
y = x.permute(2, 0, 1)
y.size()
--->[3, 64, 32]'딥러닝 > PyTorch' 카테고리의 다른 글
| PyTorch DataParallel에서 기본 GPU와 지정 GPU 불일치 문제 해결 방법 (0) | 2025.02.19 |
|---|---|
| [PyTorch] tensorboard 대충 정리 (0) | 2021.11.02 |
| [PyTorch] Pytorch.cuda.is_available() is false 해결법 (0) | 2021.10.23 |
| [Pytorch] torch.clamp (0) | 2021.10.18 |
| [Pytorch] torch.linspace (0) | 2021.10.18 |
'딥러닝/PyTorch' Related Articles
more
