반응형
TORCH.NUMEL
torch.numel(input) → int
Returns the total number of elements in the input tensor.
torch Tensor의 텐서 크기 구하기
torch.numel() 함수
>>> a = torch.randn(1, 2, 3, 4, 5)
>>> torch.numel(a)
120
>>> a = torch.zeros(4,4)
>>> torch.numel(a)
16반응형
'Boostcamp AI tech 3기 > 개념 이해' 카테고리의 다른 글
| no module named 'tensorflow' 주피터노트북 (0) | 2022.04.27 |
|---|---|
| Error loading preloads: Could not find renderer 터미널 껐다 다시 키기 (0) | 2022.04.27 |
| pytorch warning 해결 (0) | 2022.03.23 |
| unable to display MLflow UI... -> mlruns 폴더가 있는 경로에서 하자. (1) | 2022.02.16 |
| torch.range is deprecated warning -> torch.arange (0) | 2022.01.25 |
| 텐서 함수 조금 정리. 생성, 타입, select, cat, stack (0) | 2022.01.25 |
| pytorch의 reshape and view 차이 (+ contiguity 문제) (0) | 2022.01.24 |
| [Week02] Pytorch - 3강 Pytorch 기본 환경 세팅 : colab에서 ssh 연결해 vscode 실행하기 (0) | 2022.01.24 |