torch.nn
在實際使用中,最常見的做法是繼承nn.module,撰寫自己的網路層。nn.module
torch.nn.functional.pad()
padding操作
padding操作是給影象外圍加畫素點。
具體:pytorch 中pad函式toch.nn.functional.pad()的使用
detach()
將variable引數從網路中隔離開,不參與引數更新
permute(dims)
將tensor的維度換位,假設img的維度為(28,28,3), img.permute(2,0,1)得到乙個(3,28,28)的tensor.
torch.nn.conv2d(in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=true)
卷積。
PyTorch 深度學習 筆記
方差 偏差 線性回歸來確定兩種或兩種以上變數間相互依賴的定量關係。線性回歸對於輸入x和輸出y有乙個對映 類似ax b 而我們是訓練a b這兩個引數。以下 使用pytorch建立乙個線性的模型來對其進行擬合,即訓練過程。def linear ex x np.random.rand 256 noise ...
深度學習 Pytorch學習筆記(一)
pytorch中需要自己定義網路模型,該模型需要封裝到乙個自定義的類中,該類只是乙個子類,其繼承的父類為torch.nn.module,可見如下樹形結構圖 module實際又是繼承了object類,關於為什麼要繼承object類有興趣的可以看這篇部落格mro演算法 也就是說,自定義的模型除了要有 i...
深度學習 Pytorch學習筆記(五)
pytorch實現卷積神經網路 執行效率相對於keras慢太多 import torch import warnings import torchvision from torchvision.datasets import mnist from torch.utils.data import da...