docstr: Check that method `forward` produces correctly-shaped outputs.
docstr_tok: ["Check","that","method","`","forward","`","produces","correctly-shaped","outputs","."]
code: def test_output_shape_forward(self):
      r"""Check that method `forward` produces correctly-shaped outputs."""
      input_config = self._get_input_config()
      batch_size = input_config["batch_size"]
      joiner_max_input_length = input_config["joiner_max_input_length"]
      max_target_length = input_config["max_target_length"]
      num_symbols = input_config["num_symbols"]
  
      inputs, input_lengths = self._get_transcriber_input()
      targets, target_lengths = self._get_predictor_input()
  
      rnnt = self._get_model()
  
      state = None
      for _ in range(2):
          out, out_lengths, target_lengths, state = rnnt(
              inputs, input_lengths, targets, target_lengths, state
          )
          self.assertEqual(
              (
                  batch_size,
                  joiner_max_input_length,
                  max_target_length,
                  num_symbols,
              ),
              out.shape,
          )
          self.assertEqual((batch_size,), out_lengths.shape)
          self.assertEqual((batch_size,), target_lengths.shape)
  
file_path: audio/test/torchaudio_unittest/prototype/rnnt_test_impl.py
imports: import torch
  from torchaudio.prototype.models import conformer_rnnt_model
  from torchaudio_unittest.common_utils import TestBaseMixin, torch_script
label_desc: Use image node_id 12 for calling the ConformerRNNTTestImpl obj's underlying member method code with example usage: obj.test_output_shape_forward() without return types