*numel

- n = numel(A)는 배열 A의 요소 개수 n을 반환.

- prod(size(A))와 동일

 

 

반응형

'스타트업 > AI' 카테고리의 다른 글

[AI] GAN  (0) 2020.02.10
[AI] autograd, torchvision.transforms  (0) 2020.02.05
[AI] squeeze(), unsqueeze()  (0) 2020.02.04
[AI] Mutil-GPU(in Pytorch)  (0) 2020.02.04
[AI] cudnn.benchmark = True  (0) 2020.02.03

+ Recent posts