src/transformers/models/poolformer/modeling_poolformer.py/PoolFormerPooling/forward
class PoolFormerPooling:    def forward(self, hidden_states):
        return self.pool(hidden_states) - hidden_states