[ 인공지능/프레임워크 or 라이브러리 ]
파이토치 개요, 텐서의 형변환, 연산, 함수
2024-02-10 15:31:38
설 연휴라 집에 내려와서 노트북으로 작성 중이다 노트북이 오래돼서 그런지 너무 느려서 예전에 포맷을 했었는데 그 이후로는 나름 쓸만한 것 같아서 다행이다 이번에는 파이썬에 있는 딥러닝을 위한 프레임워크인 파이토치에 대해서 알아보고 정리해볼 것이다 가보자고~ 1. 파이토치(PyTorch) 개요 PyTorch는 기계 학습 프레임워크(framework) 중 하나다 PyTorch를 사용하면, GPU 연동을 통해 효율적으로 딥러닝 모델을 학습할 수 있다 import torch 파이토치를 사용하여 연산을 할 때 CPU를 사용할 것인지 GPU를 사용할 것인지를 정할 수 있다 data = [ [1, 2], [3, 4] ] x = torch.tensor(data) print(x.is_cuda) x = x.cuda() #..