雷锋网 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 链接:
https://github.com/pytorch/pytorch/releases/tag/v0.4.0
PyTorch 官网链接:
http://pytorch.org/
相关文章: