docstr: Returns the last layer hidden-state of the first token of of the
  sequence, the classification (cls) token.
  
  Args:
      last_hidden_state (torch.Tensor): Sequence of hidden-state of
      at the output of the last layer of the model (bs, seq length, hidden size)
  
  Returns:
      [torch.Tensor]: First token of the last hidden-state. (bs, hidden size)
docstr_tok: ["Returns","the","last","layer","hidden-state","of","the","first","token","of","of","the","sequence",",","the","classification","(","cls",")","token",".","Args",":","last_hidden_state","(","torch.Tensor",")",":","Sequence","of","hidden-state","of","at","the","output","of","the","last","layer","of","the","model","(","bs",",","seq","length",",","hidden","size",")","Returns",":","[","torch.Tensor","]",":","First","token","of","the","last","hidden-state",".","(","bs",",","hidden","size",")"]
code: def forward(self, last_hidden_state: torch.Tensor):
      """Returns the last layer hidden-state of the first token of of the
      sequence, the classification (cls) token.
  
      Args:
          last_hidden_state (torch.Tensor): Sequence of hidden-state of
          at the output of the last layer of the model (bs, seq length, hidden size)
  
      Returns:
          [torch.Tensor]: First token of the last hidden-state. (bs, hidden size)
      """
      return last_hidden_state.select(
          dim=self.dim, index=self.cls_index
      )
  
imports: from typing import Any, List
  import torch
  import torch.nn
  from mmf.common.registry import registry
label_desc: Use image node_id 2 for calling the ClsPooler obj's underlying member method code with example usage: obj.forward(last_hidden_state) and returns: last_hidden_state