import torch.nn as nn
import torch.nn.functional as F
class Model(nn.Module):
def __init__(self):
super().__init__()
self.conv1 = nn.Conv2d(1, 20, 5)
self.conv2 = nn.Conv2d(20, 20, 5)
def forward(self, x):
x = F.relu(self.conv1(x))
return F.relu(self.conv2(x))
输入x经过一次卷积,再经过一次非线性,再经过一次卷积,再经过一次非线性,才得到输出
import torch
from torch import nn
class XuZhenyu(nn.Module):
def __init__(self, *args, **kwargs) -> None:
super().__init__(*args, **kwargs)
def forward(self,input):
output = input+1
return output
xzy = XuZhenyu()
x = torch.tensor(1.0)
output = xzy(x)
print(output)
结果:
debug调试