您的位置:首頁>正文

PyTorch更新了:支持Windows,新增零維張量

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

— 完 —

誠摯招聘

同類文章
Next Article
喜欢就按个赞吧!!!
点击关闭提示