学习笔记(01):60分钟带你学习PyTorch框架。-PyTorch使用介绍

立即学习:https://edu.csdn.net/course/play/23740/267285?utm_source=blogtoedu

pytorch会自动创建计算图

pytorch转换成numpy数组直接使用

张量.numpy()

如果要将张量转到GPU上可以使用张量.cuda()

b=a.cuda() 就可以将张量从cpu转到GPU上面去

a.cpu() 就转换到cpu上去

当不知道处理器是cpu还是gpu时

b.to(a.device) 将b转换到a设备上面

只有在相同设备上面才可以进行互操作运算的

  •  

 

©️2020 CSDN 皮肤主题: 大白 设计师:CSDN官方博客 返回首页