class_docstr: A Loss module that computes loss for Tacotron2
code: def input_types(self):
      return {
          "spec_pred_dec": NeuralType(
              ("B", "D", "T"), MelSpectrogramType()
          ),
          "spec_pred_postnet": NeuralType(
              ("B", "D", "T"), MelSpectrogramType()
          ),
          "gate_pred": NeuralType(("B", "T"), LogitsType()),
          "spec_target": NeuralType(
              ("B", "D", "T"), MelSpectrogramType()
          ),
          "spec_target_len": NeuralType(("B"), LengthsType()),
          "pad_value": NeuralType(),
      }
  
file_path: NeMo/nemo/collections/tts/losses/tacotron2loss.py
imports: import torch
  from nemo.collections.tts.parts.utils.helpers import get_mask_from_lengths
  from nemo.core.classes import Loss, typecheck
  from nemo.core.neural_types.elements import LengthsType, LogitsType, LossType, MelSpectrogramType
  from nemo.core.neural_types.neural_type import NeuralType
label_desc: Use image node_id 1 for calling the Tacotron2Loss obj's underlying member method code with example usage: obj.input_types() and returns: dict