import torch.nn as nn import torch.nn.functional as F import torch.optim as optim from torch.autograd import Variable import torch class Net(nn.Module): # 需要继承这个类 def __init__(self): super(Net, self).__init__() # 建立了两个卷积层,self.conv1, self.conv2,注意,这些…
torch.nn只接受mini-batch的输入,也就是说我们输入的时候是必须是好几张图片同时输入. 例如:nn. Conv2d 允许输入4维的Tensor:n个样本 x n个色彩频道 x 高度 x 宽度 #coding=utf-8 import torch import torch.nn as nn import torch.nn.functional as F from torch.autograd import Variable class Net(nn.Module): #定义Net的初…