Pytorch之卷积层实战

import torchvisionimport torchfrom torch import nnfrom torch.nn import Conv2dfrom torch.utils.data import DataLoaderfrom torch.utils.tensorboard import SummaryWriterdataset=torchvision.datasets.CIFAR10('../Data',train=False,transform=torchvision.transforms.ToTensor(),download=True)#dataloaderdataloader=DataLoader(dataset,batch_size=64)#神经网络#torch.nn.Conv2d(in_channels, out_channels, kernel_size(卷积核), stride=1, padding=0(扩充),dilation=1, groups=1(常为一),# bias=True(偏置), padding_mode='zeros', device=None, dtype=None)class Sjnet(nn.Module):    def __init__(self):        super(Sjnet, self).__init__()#继承父类        self.conv1=Conv2d(in_channels=3,out_channels=6,kernel_size=3,stride=1,padding=0,)#卷积层    def forward(self,x):        x=self.conv1(x)        return xsjnet=Sjnet()#tensorboard展示图象writer=SummaryWriter('logs')i=0for data in dataloader:    imgs,targets=data    output=sjnet(imgs)    print(output.shape)#输出卷积后的图片信息    writer.add_images('input', imgs, i)#展示没有卷积之前的图像    output=torch.reshape(output,(-1,3,30,30))#修改卷积后的图像    writer.add_images('output',output,i)#展示卷积后的图像    i=i+1writer.close()

input:

output:

值���注意的是，卷积层的输入数据张量的维度应该符合卷积层的要求。一般来说，输入数据的维度为(batch_size, in_channels, height, width)，其中batch_size表示批次大小，in_channels表示输入的通道数，height和width表示输入数据的高度和宽度。

卷积层的输出数据张量的维度为(batch_size, out_channels, output_height, output_width)，其中out_channels表示输出的通道数，output_height和output_width表示输出数据的高度和宽度，这些维度的计算取决于输入数据的维度、卷积核的大小、步长和填充等参数。