batchnorm2d参数 torch_科学网-Pytorch中nn.Conv1d、Conv2D与BatchNorm1d、BatchNorm2d函数-张伟的博文...

此处mark一下图像处理中常见的Conv和BatchNorm操作，提供参考链接！！

(一)Conv1D和Conv2D实现

实现代码和相应卷积的数据输入格式要求

import os

import torch

import torch.nn as nn

import torch.nn.functional as F

import torch.utils.data

class myCNN(torch.nn.Module):

def __init__(self):

super(myCNN, self).__init__()

#nn.Model.__init__(self)

self.conv1 = nn.Conv2d(1, 6, 5)  # 输入通道数为1，输出通道数为6

self.con