PyTorch更新了：支持Windows，新增零维张量

李林 编译整理
量子位 出品 | 公众号 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

作者系网易新闻·网易号“各有态度”签约作者

— 完 —

活动报名

加入社群

量子位AI社群16群开始招募啦，欢迎对AI感兴趣的同学，加小助手微信qbitbot6入群；

此外，量子位专业细分群(自动驾驶、CV、NLP、机器学习等)正在招募，面向正在从事相关领域的工程师及研究人员。

进群请加小助手微信号qbitbot6，并务必备注相应群的关键词~通过审核后我们将邀请进群。（专业群审核较严，敬请谅解）

诚挚招聘

量子位正在招募编辑/记者，工作地点在北京中关村。期待有才气、有热情的同学加入我们！相关细节，请在量子位公众号(QbitAI)对话界面，回复“招聘”两个字。

量子位 QbitAI · 头条号签约作者

վ'ᴗ' ի 追踪AI技术和产品新动态