src/transformers/models/van/modeling_van.py/VanLayerScaling/forward
class VanLayerScaling:    def forward(self, hidden_state: torch.Tensor) -> torch.Tensor:
        # unsqueezing for broadcasting
        hidden_state = self.weight.unsqueeze(-1).unsqueeze(-1) * hidden_state
        return hidden_state