学习笔记(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上去 当不知道处
相关文章
相关标签/搜索