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
— 完 —
誠摯招聘