Deep Learning: Pytorch隨筆

1.torch.randn(1, 3, 3, 3)
生成隨機(jī)數(shù)向量

2.torch.Tensor.view(3, 9)
變換維度,如把上面的四維 1 * 3 * 3 * 3 變成二維 3 * 9 ,方式就是順序排列下來(lái)
view(-1, 3) 會(huì)把數(shù)據(jù)變成 n * 3 維度

3.nn.Linear(in_features = 3 * 3, out_features = 1)
全連接層,輸入必須是二維
input是 (3, 9)的情況下,in_features是9,output是 (3, 1)

4.torch.cat((A, B), dim = 1)
拼接兩個(gè)Tensor,dim = 0 豎著拼,dim = 1 橫著拼

5.nn.module
定義自己的網(wǎng)絡(luò)

需要繼承nn.Module類,并實(shí)現(xiàn)forward方法。
一般把網(wǎng)絡(luò)中具有可學(xué)習(xí)參數(shù)的層放在構(gòu)造函數(shù)init()中,
不具有可學(xué)習(xí)參數(shù)的層(如ReLU)可放在構(gòu)造函數(shù)中,也可不放在構(gòu)造函數(shù)中(而在forward中使用nn.functional來(lái)代替)

只要在nn.Module的子類中定義了forward函數(shù),backward函數(shù)就會(huì)被自動(dòng)實(shí)現(xiàn)(利用Autograd)。
在forward函數(shù)中可以使用任何Variable支持的函數(shù),畢竟在整個(gè)pytorch構(gòu)建的圖中,是Variable在流動(dòng)。還可以使用if,for,print,log等python語(yǔ)法.

注:Pytorch基于nn.Module構(gòu)建的模型中,只支持mini-batch的Variable輸入方式,
比如,只有一張輸入圖片,也需要變成 N x C x H x W 的形式:

input_image = torch.FloatTensor(1, 28, 28)
input_image = Variable(input_image)
input_image = input_image.unsqueeze(0) # 1 x 1 x 28 x 28

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡(jiǎn)書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

  • 今天感恩節(jié)哎,感謝一直在我身邊的親朋好友。感恩相遇!感恩不離不棄。 中午開(kāi)了第一次的黨會(huì),身份的轉(zhuǎn)變要...
    余生動(dòng)聽(tīng)閱讀 10,869評(píng)論 0 11
  • 彩排完,天已黑
    劉凱書法閱讀 4,482評(píng)論 1 3
  • 沒(méi)事就多看看書,因?yàn)楦褂性?shī)書氣自華,讀書萬(wàn)卷始通神。沒(méi)事就多出去旅游,別因?yàn)闆](méi)錢而找借口,因?yàn)橹灰闶〕詢€用,來(lái)...
    向陽(yáng)之心閱讀 4,974評(píng)論 3 11
  • 表情是什么,我認(rèn)為表情就是表現(xiàn)出來(lái)的情緒。表情可以傳達(dá)很多信息。高興了當(dāng)然就笑了,難過(guò)就哭了。兩者是相互影響密不可...
    Persistenc_6aea閱讀 129,698評(píng)論 2 7

友情鏈接更多精彩內(nèi)容