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

研发管理 专栏收录该内容
1 篇文章 0 订阅

立即学习: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设备上面

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

  •  

 

  • 0
    点赞
  • 0
    评论
  • 0
    收藏
  • 一键三连
    一键三连
  • 扫一扫,分享海报

©️2021 CSDN 皮肤主题: 大白 设计师:CSDN官方博客 返回首页
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、C币套餐、付费专栏及课程。

余额充值