본문 바로가기

AI/PyTorch

[PyTorch] package

1. Package load먼저, 필요한 패키지들을 로드합니다. 각각의 패키지들의 대략적인 사용목적은 다음과 같습니다.

 

  • torch: PyTorch 패키지를 불러옵니다.
  • torchvision: PyTorch 에서 이미지 데이터 로드와 관련된 여러가지 편리한 함수들을 제공합니다.
  • matplotlib.pyplot: 데이터 시각화를 위해 사용합니다.
  • numpy: Scientific computing과 관련된 여러 편리한 기능들을 제공해주는 라이브러리입니다

'AI > PyTorch' 카테고리의 다른 글

[PyTorch] Linear Regression 작성하기  (0) 2021.06.01
[PyTorch] Tensor와 NumPy  (0) 2021.06.01
[pyTorch] 이미지 여는 방법  (0) 2021.05.12
[PyTorch] torch view, squeeze, unsqueeze  (0) 2021.04.27
[PyTorch] torch 함수  (0) 2021.04.27