雷锋网 AI 研习社最新消息，PyTorch 官方发布 0.4.0 版本，该版本的 PyTorch 有多项重大更新，其中最重要的改进是支持 Window 系统。

2017 年初，Facebook 在机器学习和科学计算工具 Torch 的基础上，针对 Python 语言发布了全新的机器学习工具包——PyTorch。PyTorch 实际上是 NumPy 的替代，它支持 GPU，有着更高级的功能，可以用来构建和训练深度神经网络。一经发布，它便受到广泛关注和讨论。经过将近一年的发展，目前 PyTorch 已经成为从业者最重要的研发工具之一。

以下为 PyTorch 0.4.0 版本主要更新内容：

主要核心改变

Tensor/Variable 合并

零维张量

dtypes

迁移指导

新特性

Tensor

全面支持高级索引

快速傅里叶变换

神经网络

权衡内存计算

bottleneck -- 识别你代码中热点的工具

torch.distributions

24 基本的概率分布

增加了 cdf, variance, entropy, perplexity 等

分布式训练

易于使用的 Launcher 程序

NCCL2 后端

C++ 扩展

支持 Windows

ONNX 改进

支持 RNN

性能改善

Bug 修复

详细信息请查看 PyTorch 的 Github 库以及官方主页。

Github 链接：

PyTorch 官网链接：