torch.nn的核心資料結構是module，既可以表示神經網路中的某個層（layer），也可以表示乙個包含很多層的神經網路。
import torch as t
from torch import nn
主要關注以下幾點
輸入的不是單個資料，而是乙個batch。輸入只有乙個資料，則必須呼叫tensor.unsqueeze(0)或tensor[none]將資料
pytorch 使用預訓練層
將其他地方訓練好的網路，用到新的網路裡面 1.原先已經訓練好乙個網路 autoencoder fc 2.首先載入該網路，讀取其儲存的引數 3.設定乙個引數集 cnnpre autoencoder fc cnnpre.load state dict torch.load autoencoder fc....
PyTorch 自定義層
與使用module類構造模型類似。下面的centeredlayer類通過繼承module類自定義了乙個將輸入減掉均值後輸出的層，並將層的計算定義在了forward函式裡。這個層裡不含模型引數。class mydense nn.module def init self super mydense,se...
PyTorch入門（三）PyTorch常用操作
def bilinear kernel in channels,out channels,kernel size return a bilinear kernel tensor tensor in channels,out channels,kernel size,kernel size 返回雙線性...