class_docstr: From BertOnlyMLMHead, BertLMPredictionHead, BertPredictionHeadTransform
class_docstr_tok: ["From","BertOnlyMLMHead",",","BertLMPredictionHead",",","BertPredictionHeadTransform"]
code: def forward(self, unpooled):
      hidden_states = self.dense(unpooled)
      hidden_states = self.transform_act_fn(hidden_states)
      hidden_states = self.LayerNorm(hidden_states)
      logits = self.decoder(hidden_states) + self.bias
      return logits
  
imports: from __future__ import annotations
  import abc
  import torch
  import torch.nn
  import transformers
  from jiant.ext.allennlp import SelfAttentiveSpanExtractor
  from jiant.shared.model_resolution import ModelArchitectures
  from jiant.tasks.core import TaskTypes
  from typing import Callable
  from typing import List
label_desc: Use image node_id 2 for calling the BertMLMHead obj's underlying member method code with example usage: obj.forward(unpooled) and returns: logits