【PyTorch】使用 MaxPool 实现图像的膨胀和腐蚀

PyTorch使用 MaxPool 实现图像的膨胀和腐蚀：

max_pool = nn.MaxPool2d(kernel_size=5, stride=1, padding=2)  # 可调整 kernel_size 和 padding
# 假设要处理的张量为 tensor
# ===== 膨胀 =====
tensor_dilate = max_pool(tensor)
# ===== 腐蚀 =====
tensor_erode = -max_pool(-tensor)

————————————————
来源：https://blog.csdn.net/qq_39546227/article/details/122473559