StochasticPooling 随机池化的PyTorch实现

网上找了好久随机池化的Pytorch代码，都没有太合适的。就自己写了个不太优美的实现。若是在论文中使用该代码，请私信告知我用途。

from torch import nn
import torch
from torch.nn import functional as F

class StochasticPooling(nn.Module):
    def __init__(self):
        super().__init__()
        self.avg = nn.AvgPool2d(2)

    def forward(self, x):
        s = 4