上传者: 38670949
|
上传时间: 2023-03-03 14:46:01
|
文件大小: 67KB
|
文件类型: PDF
3*3卷积核与2*5卷积核对神经元大小的设置
#这里kerner_size = 2*5
class CONV_NET(torch.nn.Module): #CONV_NET类继承nn.Module类
def __init__(self):
super(CONV_NET, self).__init__() #使CONV_NET类包含父类nn.Module的所有属性
# super()需要两个实参,子类名和对象self
self.conv1 = nn.Conv2d(1, 32, (2, 5), 1, padding=0)
self.conv2 = nn.Conv2d(32, 1