머신러닝
[pytorch] 파이토치 텐서 자료형
코딩하는 람지
2022. 9. 16. 18:34
32비트의 부동 소수점: torch.FloatTensor
64비트의 부호 있는 정수: torch.LongTensor
GPU 연산을 위한 자료형: torch.cuda.FloatTensor