1、数据集:CIFAR10
2、二维最大池化MaxPool2d可视化
1、相关参数
2、代码及说明
代码
import torch
import torchvision
from torch import nn
from torch.nn import MaxPool2d
from torch.utils.data import DataLoader
from torch.utils.tensorboard import SummaryWriter
#使用CIFAR10数据集
dataset = torchvision.datasets.CIFAR10("E:/PycharmProjects/Pytoch_learning/dataset/CIFAR10",
train=False