PyTorch基礎

紅圈圈裏的數據類型比較重要 瞎貼一波： 一、數據類型 a = torch.randn(2,3) print(a) print(type(a)) print(isinstance(a, torch.FloatTensor))#合法化檢驗 print(isinstance(a, torch.cuda.FloatTensor))#合法化檢驗 a = a.cuda()#返回一個GPU上的引用 print

>>阅读原文<<