PyTorch 起步走(I)

Yen-Ying Lee
2 min readOct 17, 2017

--

PyTorch 簡介

PyTorch 是一個由FAIR所推出的Deep Learning 框架,從它的名字可以看出來與原本Torch有血緣關係,但是將其優點融合進去原本的Python之中,讓你用起來就像是使用numpy/scikit-learn一樣,不會有種學習一個全新的框架的感覺。

安裝

安裝PytTorch是件十分簡單的事情,只要到官網上點選自己的系統和Python版本,下面就會自動帶出安裝的指令,可説是十分貼心。

Tensor

Tensor是PyTorch裏面基本且重要的元素,它就像numpy中的ndarrays,但是PyTorch中加上了GPU的支援,可以讓計算來的更加快速。

上面就是兩個常見簡單建立Tensor的方法,其他方法在官網的文件有更多的介紹,大家有興趣可以去看看。

操作Tensor

當你要去做一些Tensor的計算時,一定會需要知道要如何操作它。下面先用加法當作例子,展示一些加法的各式寫法。

這個文件裏面有著更詳細的運算符號的介紹,大家可以參考看看。

然而Tensor内的元素取值方法,則是跟numpy完全一致,如果你是熟悉numpy的朋友,完全就是可以無縫接軌呢:)

經過了上面簡單的介紹,相信大家對於PyTorch應該有基本的瞭解了,下一篇文章會用PyTorch來帶大家實作Linear Regression,請大家敬請期待 :)

--

--