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

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

原文发布于微信公众号 - 量子位(QbitAI)

原文发表时间:2018-04-25

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏华章科技

做数据分析必须学R的4个理由

R 是一种灵活的编程语言,专为促进探索性数据分析、经典统计学测试和高级图形学而设计。R 拥有丰富的、仍在不断扩大的数据包库,处于统计学、数据分析和数据挖掘发展的...

611
来自专栏PPV课数据科学社区

【连载】R语言18讲(四)

? 前面讲到了怎么导入数据,获取数据,按照数据挖掘的流程,在我们导入数据之后接下来的工作便是对数据进行初步的探索,探索的工作主要包括两大块,第一是数据...

2868
来自专栏IT派

算法应用实践:如何用Python写一个贪吃蛇AI

这两天在网上看到一张让人涨姿势的图片,图片中展示的是贪吃蛇游戏, 估计大部分人都玩过。但如果仅仅是贪吃蛇游戏,那么它就没有什么让人涨姿势的地方了。 问题的关键在...

1980
来自专栏生信小驿站

Python Vs R:数据科学家的永恒问题pythonR结论

Python有一些使用案例,R也是如此。使用它们的场景各不相同。 更常见的是环境以及客户或雇主的需求决定了Python和R之间的选择。许多事情在Python中都...

912
来自专栏量子位

有个AI陪你一起写代码,是种怎样的体验?| 附ICLR论文

后来,程序猿要写的代码越来越多,世界上便有了各种各样的API,来减少大家的工作量。有些功能,可以让API来帮我们实现。

1002
来自专栏PPV课数据科学社区

做数据分析必须学R语言的4个理由

R 是一种灵活的编程语言,专为促进探索性数据分析、经典统计学测试和高级图形学而设计。R 拥有丰富的、仍在不断扩大的数据包库,处于统计学、数据分析和数据挖掘发展的...

3766
来自专栏小文博客

用C语言写贪吃蛇笔记-1

3193
来自专栏PPV课数据科学社区

告诉你做数据分析必须学R的4个理由

论坛君:你很可能已经听说过 R,或许你知道 R 是一种编程语言,而且知道它与统计学有关,但它是否适合您呢?本文作者将试图向大家讲解他对R的看法,分享他认为试用开...

3406
来自专栏哲学驱动设计

重构实践:体验interface的威力(一)

背景     GIX4是一个建筑行业的指标计算软件,用于数据统计、分析。导入的大量数据,大部分呈现逻辑上的树状结构(关于它的重构,见:《重构一个繁琐的数据结构》...

1977
来自专栏AI研习社

谷歌云工程师亲自示范:新发布的 Video Intelligence API 究竟怎么用

AI研习社按:3 月初,谷歌在 Google Cloud Next 2017 大会上对外公布了 Cloud Machine Learning API 更新。此次...

3347

扫码关注云+社区