docstr: Run the Transformer forward and check output shape.
docstr_tok: ["Run","the","Transformer","forward","and","check","output","shape","."]
code: def test_transformer_forward_shape(
      self, input_vocab_size, output_vocab_size
  ):
      """Run the Transformer forward and check output shape."""
      self._test_transformer_forward_shape(
          input_vocab_size, output_vocab_size
      )
  
file_path: trax/trax/models/transformer_test.py
imports: import functools
  from absl.testing import absltest
  from absl.testing import parameterized
  import numpy
  from trax import fastmath
  from trax import shapes
  from trax.layers import test_utils
  from trax.models import transformer
label_desc: Use image node_id 3 for calling the TransformerTest obj's underlying member method code with example usage: obj.test_transformer_forward_shape(input_vocab_size, output_vocab_size) without return types