src/transformers/models/regnet/modeling_regnet.py/RegNetSELayer/forward
class RegNetSELayer:    def forward(self, hidden_state):
        # b c h w -> b c 1 1
        pooled = self.pooler(hidden_state)
        attention = self.attention(pooled)
        hidden_state = hidden_state * attention
        return hidden_state