class_docstr: Neural module which encodes BERT hidden states
docstr: Args:
      hidden_states: bert output hidden states
docstr_tok: ["Args",":","hidden_states",":","bert","output","hidden","states"]
code: def forward(self, hidden_states):
      """
      Args:
          hidden_states: bert output hidden states
      """
      first_token_hidden_states = hidden_states[:, 0]
      logits = self.fc(first_token_hidden_states)
      logits = self.activation(logits)
      logits = self.dropout1(logits)
      return logits, self.dropout2(hidden_states)
  
file_path: NeMo/nemo/collections/nlp/modules/dialogue_state_tracking/sgd_encoder.py
imports: from typing import Dict, Optional
  from torch import nn
  from nemo.collections.nlp.modules.common.classifier import Classifier
  from nemo.core.classes import typecheck
  from nemo.core.neural_types import ChannelType, LogitsType, NeuralType
label_desc: Use image node_id 3 for calling the SGDEncoder obj's underlying member method code with example usage: obj.forward(hidden_states) and returns: logits,  self