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

>>阅读原文<<