pytorch中池化层MaxPool2d函数的stride参数

class torch.nn.MaxPool2d(kernel_size, stride=None, padding=0, dilation=1, return_indices=False, ceil_mode=False)

kernel_size(int or tuple) - max pooling的窗口大小
stride(int or tuple, optional) - max pooling的窗口移动的步长。默认值是kernel_size

你可能感兴趣的:(pytorch中池化层MaxPool2d函数的stride参数)