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

作者系網易新聞·網易號“各有態度”簽約作者

— 完 —

活動報名

加入社群

量子位AI社群16群開始招募啦，歡迎對AI感興趣的同學，加小助手微信qbitbot6入群；

此外，量子位專業細分群(自動駕駛、CV、NLP、機器學習等)正在招募，面向正在從事相關領域的工程師及研究人員。

進群請加小助手微訊號qbitbot6，並務必備註相應群的關鍵詞~通過稽核後我們將邀請進群。（專業群稽核較嚴，敬請諒解）

誠摯招聘

量子位正在招募編輯/記者，工作地點在北京中關村。期待有才氣、有熱情的同學加入我們！相關細節，請在量子位公眾號(QbitAI)對話介面，回覆“招聘”兩個字。

量子位 QbitAI · 頭條號簽約作者

վ'ᴗ' ի 追蹤AI技術和產品新動態