src/transformers/models/mobilebert/modeling_mobilebert.py/MobileBertLMPredictionHead/forward
class MobileBertLMPredictionHead:    def forward(self, hidden_states: torch.Tensor) -> torch.Tensor:
        hidden_states = self.transform(hidden_states)
        hidden_states = hidden_states.matmul(torch.cat([self.decoder.weight.t(), self.dense.weight], dim=0))
        hidden_states += self.decoder.bias
        return hidden_states