class_docstr: Base class to calculate hidden state loss.
  Returned dict includes a loss value and additional outputs.
class_docstr_tok: ["Base","class","to","calculate","hidden","state","loss",".","Returned","dict","includes","a","loss","value","and","additional","outputs","."]
docstr: Returns and caches input names
docstr_tok: ["Returns","and","caches","input","names"]
code: def input_names(self):
      """Returns and caches input names"""
      # we always expect hiddens_mask to be used to mask out loss of padded elements
      return self._input_names() + ["hiddens_mask"]
  
imports: import math
  import torch
label_desc: Use image node_id 4 for calling the MegatronBaseHiddenLoss obj's underlying member method code with example usage: obj.input_names() and returns: unknown