머신러닝

[pytorch] 파이토치 텐서 자료형

코딩하는 람지 2022. 9. 16. 18:34

32비트의 부동 소수점: torch.FloatTensor

64비트의 부호 있는 정수: torch.LongTensor

GPU 연산을 위한 자료형: torch.cuda.FloatTensor