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怎么使用”有了更深的了解,不妨来实际操作一番吧!这里是亿速云网站,更多相关内容可以进入相关频道进行查询,关注我们,继续学习!