nn.MaxPool2d()

nn.MaxPool2d() 是 PyTorch 中用于创建最大池化层的函数。最大池化层通常用于减小图像或特征图的空间尺寸，同时保留重要的特征。该函数的常用参数包括 kernel_size（池化窗口大小）、stride（步长）、padding（填充）等。以下是一个示例：

import torch.nn as nn

# 创建一个最大池化层
max_pool_layer = nn.MaxPool2d(kernel_size=2, stride=2)

# 假设有一个输入张量 input_tensor，其形状为 [batch_size, channels, height, width]
# 经过最大池化层后，输出张量的形状为 [batch_size, channels, height/2, width/2]
output_tensor = max_pool_layer(input_tensor)