torch.nn
- Containers
- Convolution Layers
- Pooling layers
- Padding Layers
- Non-linear Activations (weighted sum, nonlinearity)
- Non-linear Activations (other)
- Normalization Layers
- Recurrent Layers
- Transformer Layers
- Linear Layers
- Dropout Layers
- Sparse Layers
- Distance Functions
- Loss Functions
- Vision Layers
- Shuffle Layers
- DataParallel Layers (multi-GPU, distributed)
- Utilities
- Quantized Functions
- Lazy Modules Initialization
모델을 만드는데 필요한 모듈을 제공한다.
torch.nn — PyTorch 1.8.1 documentation
Shortcuts
pytorch.org
torch.nn.functional
- Convolution functions
- Pooling functions
- Non-linear activation functions
- Normalization functions
- Linear functions
- Dropout functions
- Sparse functions
- Distance functions
- Loss functions
- Vision functions
- DataParallel functions (multi-GPU, distribut
torch.nn.functional — PyTorch 1.8.1 documentation
Shortcuts
pytorch.org
'AI > PyTorch' 카테고리의 다른 글
| [PyTorch] CNN 설계 4. Dataset 및 DataLoader 할당 (0) | 2021.06.15 |
|---|---|
| [PyTorch] torch.optim의 인자 model.parameters() (0) | 2021.06.15 |
| [PyTorch] CNN 설계 순서 1 - 3 (0) | 2021.06.15 |
| [PyTorch] torch.nn 과 torch.nn.functional (0) | 2021.06.10 |
| [PyTorch] nn.MaxPool2d(), nn.BatchNorm2d() (0) | 2021.06.10 |
