日常筆記 pytorch深度學習常用函式

2021-09-27 10:16:21 字數 492 閱讀 6895

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...