立即学习: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设备上面
只有在相同设备上面才可以进行互操作运算的
423
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
