pytorch cnn pool bn 等速查表

1. nn.Conv2d

nn.Conv2d(int_c, out_c, kernel-size=(3,3),  stride = (1,1),  padding=(1,1))

2. nn.BatchNorm2d

nn.BatchNorm2d(num_feature)  #特征图数目

3. F.avg_pool2d

F.avg_pool2d( input_data,  kernel-size)