专栏首页量子位PyTorch更新了:支持Windows,新增零维张量

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),作者:关注前沿科技

原文出处及转载信息见文内详细说明,如有侵权,请联系 yunjia_community@tencent.com 删除。

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

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

我来说两句

0 条评论
登录 后参与评论

相关文章

  • FB发布PyTorch:一个GPU加速的Python深度学习框架

    作者 | 量子位 舒石 经过数月的工作,Facebook今天终于对外发布PyTorch的Beta版本。PyTorch是一个Python优先的深度学习框架。先来看...

    量子位
  • PyTorch发布一周年:盘点社区和工程大事件,后来者居上态势已显?

    安妮 编译自 PyTorch官网 今天,PyTorch公开发布一周年了。 这个自诞生之日起就火热的深度学习框架,这一年是怎样的成长轨迹?刚刚,PyTorch团...

    量子位
  • TensorFlow王位不保?ICLR投稿论文PyTorch出镜率快要反超了

    如今,PyTorch 1.0发布,ICLR 2019也才截稿不久,又是讨论这个问题的好时节。

    量子位
  • PyTorch 1.6、TensorFlow 2.3、Pandas 1.1同日发布!都有哪些新特性?

    7月29日,PyTorch 1.6、TenorFlow 2.3、Pandas 1.1恰巧同时发布。这三个库都是定期滚动更新,大约每个季度更新一次小版本。

    AI科技大本营
  • PyTorch 到底好用在哪里?

    提问内容如下: 之前非常熟悉 Tensorflow,后来都说 PyTorch 简单易上手,自己就去试了试。 PyTorch 连最基本的 maximum, min...

    AI研习社
  • 一行代码切换TensorFlow与PyTorch,模型训练也能用俩框架

    在早两天开源的 TfPyTh 中,不论是 TensorFlow 还是 PyTorch 计算图,它们都可以包装成一个可微函数,并在另一个框架中高效完成前向与反向传...

    机器之心
  • 一行代码切换TensorFlow与PyTorch,模型训练也能用俩框架

    在早两天开源的 TfPyTh 中,不论是 TensorFlow 还是 PyTorch 计算图,它们都可以包装成一个可微函数,并在另一个框架中高效完成前向与反向传...

    代码医生工作室
  • EasyPlayer网页全终端播放器之Android版的分屏策略及如何设置2分屏

    要实现这几种显示模式。其实只要对播放控件的布局进行些许调整即可。**那EasyPlayer-Android的分屏策略和二分屏设置是如何实现呢?

    EasyNVR
  • 【坑】时效性数据传参的后果

    版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.n...

    林老师带你学编程
  • 硬刚 Tensorflow 2.0 ,PyTorch 1.3 今日上线!

    就在今年 8 月份,机器学习框架 PyTorch 刚发布 1.2 版本,很多开发者甚至还没来得及吃透 1.2,两个月不到,进击的 Pytorch 又带着我们进入...

    AI科技评论

扫码关注云+社区

领取腾讯云代金券