dim 은 해당 dim을 모두 합친다. 하나의 그룹이라 생각하고 계산한다.
'AI > PyTorch' 카테고리의 다른 글
| [PyTorch] 모델 앙상블(ensemble) 하기 (0) | 2021.10.29 |
|---|---|
| [PyTorch] numpy.mean() (0) | 2021.07.12 |
| [PyTorch] CNN 설계 11. Transfer Learning (0) | 2021.07.01 |
| [PyTorch] CNN 설계 10.저장된 모델 불러오기 및 test (0) | 2021.06.29 |
| [PyTorch] torch.save(model.state_dict(), PATH) / torch.load(PATH) (0) | 2021.06.29 |