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

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