docstr: Check that method `predict` produces correctly-shaped outputs.
docstr_tok: ["Check","that","method","`","predict","`","produces","correctly-shaped","outputs","."]
code: def test_output_shape_predict(self):
      r"""Check that method `predict` produces correctly-shaped outputs."""
      input_config = self._get_input_config()
      batch_size = input_config["batch_size"]
      max_target_length = input_config["max_target_length"]
  
      model_config = self._get_model_config()
      encoding_dim = model_config["encoding_dim"]
      input, lengths = self._get_predictor_input()
  
      rnnt = self._get_model()
  
      state = None
      for _ in range(2):
          out, out_lengths, state = rnnt.predict(input, lengths, state)
          self.assertEqual(
              (batch_size, max_target_length, encoding_dim), out.shape
          )
          self.assertEqual((batch_size,), out_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 14 for calling the ConformerRNNTTestImpl obj's underlying member method code with example usage: obj.test_output_shape_predict() without return types