class_docstr: Creation of a class to combine the decoder and the lm head 
class_docstr_tok: ["Creation","of","a","class","to","combine","the","decoder","and","the","lm","head"]
code: def forward(self, input_ids, attention_mask, encoder_hidden_states):
      decoder_output = self.decoder(
          input_ids=input_ids,
          encoder_attention_mask=attention_mask,
          encoder_hidden_states=encoder_hidden_states,
      )
  
      return (
          self.lm_head(
              decoder_output[0] * (self.config.d_model**-0.5)
          ),
          decoder_output[1],
      )
  
file_path: fastT5/fastT5/onnx_models_structure.py
imports: import torch
label_desc: Use image node_id 2 for calling the DecoderWithLMheadInitial obj's underlying member method code with example usage: obj.forward(input_ids, attention_mask, encoder_hidden_states) and returns: self,  decoder_output