Deep Learning5 [Pytorch]Pytorch Basic 파이토치 모델을 가져와봐서 사용해 본 적은 몇 번 있지만 기초부터 배워보는 건 이번이 처음이다 ndarray, Tensorflow의 tensor와 다른 점 혹은 처음 보는 개념만 짚고 설명해 보겠다 Tensor 다차원 Array를 표현하는 Pytorch의 클래스이며 numpy의 ndarray, Tensorflow의 Tensor와 유사함 즉 ndarray에서 사용가능한 attribute를 대부분 이름 그대로 사용가능하다 a = torch.tensor([[1,2,3], [4,5,6]]) print(a.ndim, a.shape) 2 torch.Size([2, 3]) 텐서플로우의 텐서 혹은 ndarray와 차이점은 GPU에 올려두고 사용할 수 있다는 점 print(a.device) if torch.cuda.is.. 2023. 3. 19. 이전 1 2 다음