src/transformers/models/poolformer/modeling_poolformer.py/PoolFormerPooling/__init__
class PoolFormerPooling:    def __init__(self, pool_size):
        super().__init__()
        self.pool = nn.AvgPool2d(pool_size, stride=1, padding=pool_size // 2, count_include_pad=False)