*autograd

- autograd 패키지는 텐서의 모든 연산에 대하여 자동 미분을 제공

 

*transforms.Compose (torchvision.transforms)

- 데이터 전처리 패키지

- 타입이나 데이터 argument 하는 함수 내장

 

반응형

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

[AI] 논문 리스트  (0) 2020.02.21
[AI] GAN  (0) 2020.02.10
[AI] numel  (0) 2020.02.05
[AI] squeeze(), unsqueeze()  (0) 2020.02.04
[AI] Mutil-GPU(in Pytorch)  (0) 2020.02.04

+ Recent posts