.size() 또는 .shape 함수를 이용하여 확인한다.
'AI > Pytorch' 카테고리의 다른 글
| torch.nn.CrossEntropyLoss (0) | 2023.03.11 |
|---|---|
| Learning rate scheduler (0) | 2023.03.07 |
| Optimizer (0) | 2023.03.06 |
| DataLoader 정리 (0) | 2023.03.03 |
AI/Pytorch
.size() 또는 .shape 함수를 이용하여 확인한다.
| torch.nn.CrossEntropyLoss (0) | 2023.03.11 |
|---|---|
| Learning rate scheduler (0) | 2023.03.07 |
| Optimizer (0) | 2023.03.06 |
| DataLoader 정리 (0) | 2023.03.03 |
