李林 编译整理
量子位 出品 | 公众号 QbitAI
PyTorch今天发布了v0.4.0版本,网友们反响非常热烈,甚至有人说:感觉就像提前过圣诞~
这次的更新中,最受关注的有两点。一是Tensors(张量)和Variables(变量)的合并;二是官方终于支持Windows系统了。
以下是本次更新的主要内容:
重大核心变化:
Tensors和Variables合并了
有的运算会返回零维张量(标量)
volatile flag作废了
可以通过torch.dtype、torch.device和torch.layout更好地管理数据类型属性
支持编写不区分设备的代码
除此之外,PyTorch在张量、神经网络、分布式训练等方面增加了一些新功能,包括:
张量
全面支持高级索引
支持快速傅立叶变换
神经网络
新增了一个checkpoint容器,可以在内存使用和计算速度之间折衷
新增了一个名叫瓶颈的工具,可以对程序中的瓶颈进行初步debug
torch.distributions
扩展到包含24个基本概率分布
增加了cdf,方差,熵,复杂度等计算
分布式训练
增加了易用的启动器
提供基于NCCL2的新分布式后端
为自己编写C++或CUDA扩展提供了更好的系统
官方支持Windows,不过Windows版不支持分布式训练,可能比Linux版慢一点
改进了ONNX的支持
提供更好的RNN支持
在这次更新前不久,PyTorch还发生过一次重大变化:Caffe 2源代码全部并入了PyTorch,Facebook这大深度学习框架合二为一。
最后,要用在v0.3.x之前写的旧版代码,一定要看一眼官方迁移指南哦~
http://pytorch.org/2018/04/22/0_4_0-migration-guide.html
作者系网易新闻·网易号“各有态度”签约作者
— 完 —
活动报名
![640?wx_fmt=jpeg](https://i-blog.csdnimg.cn/blog_migrate/312667699f8ed33f6db1380fa29f75b2.jpeg)
加入社群
量子位AI社群16群开始招募啦,欢迎对AI感兴趣的同学,加小助手微信qbitbot6入群;
此外,量子位专业细分群(自动驾驶、CV、NLP、机器学习等)正在招募,面向正在从事相关领域的工程师及研究人员。
进群请加小助手微信号qbitbot6,并务必备注相应群的关键词~通过审核后我们将邀请进群。(专业群审核较严,敬请谅解)
诚挚招聘
量子位正在招募编辑/记者,工作地点在北京中关村。期待有才气、有热情的同学加入我们!相关细节,请在量子位公众号(QbitAI)对话界面,回复“招聘”两个字。
![640?wx_fmt=jpeg](https://i-blog.csdnimg.cn/blog_migrate/c3b5c4250fcc2ac2b7fad741aeff9083.jpeg)
量子位 QbitAI · 头条号签约作者
վ'ᴗ' ի 追踪AI技术和产品新动态