docstr: Check that method `join` produces correctly-shaped outputs.
docstr_tok: ["Check","that","method","`","join","`","produces","correctly-shaped","outputs","."]
code: def test_output_shape_join(self):
      r"""Check that method `join` 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"]
  
      (
          utterance_encodings,
          utterance_lengths,
          target_encodings,
          target_lengths,
      ) = self._get_joiner_input()
  
      rnnt = self._get_model()
  
      out, src_lengths, tgt_lengths = rnnt.join(
          utterance_encodings,
          utterance_lengths,
          target_encodings,
          target_lengths,
      )
      self.assertEqual(
          (
              batch_size,
              joiner_max_input_length,
              max_target_length,
              num_symbols,
          ),
          out.shape,
      )
      self.assertEqual((batch_size,), src_lengths.shape)
      self.assertEqual((batch_size,), tgt_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 15 for calling the ConformerRNNTTestImpl obj's underlying member method code with example usage: obj.test_output_shape_join() without return types