src/transformers/models/vit_mae/modeling_vit_mae.py/ViTMAEIntermediate/forward
class ViTMAEIntermediate:    def forward(self, hidden_states: torch.Tensor) -> torch.Tensor:
        hidden_states = self.dense(hidden_states)
        hidden_states = self.intermediate_act_fn(hidden_states)

        return hidden_states