PyTorch怎么使用

本篇内容主要讲解"PyTorch怎么使用"，感兴趣的朋友不妨来看看。本文介绍的方法操作简单快捷，实用性强。下面就让小编来带大家学习"PyTorch怎么使用"吧!

代码如下

import timeimport torch ##################################################for i in range(1,10):start = time.time()a = torch.FloatTensor(i*100,1000,1000)a = a.cuda() #a = a    a = torch.matmul(a,a)end = time.time() - startprint(end)

到此，相信大家对"PyTorch怎么使用"有了更深的了解，不妨来实际操作一番吧！这里是网站，更多相关内容可以进入相关频道进行查询，关注我们，继续学习！