Torch 설치 및 사용

pip install torch torchvision torchaudio
import torch

Tensor 자료형 변수 만들기

scalar = torch.tensor(3)
scalar

차원 갯수 체크

scalar.ndim

각 차원별 데이터 수 체크

scalar.shape

데이터 타입 체크

scalar.dtype

Tensor 변수가 현재 할당된 장치 체크

scalar.dtype

Tensor 변수 내부의 값 꺼내기