code: def test_style_based_generator_mapping_outputs_shape(
      self, batch_size
  ):
      input_data = tf.ones(shape=(batch_size, 512))
      output_dimension = 554
      mapping_network = architectures_style_gan.create_mapping_network(
          latent_code_dimension=512,
          output_dimension=output_dimension,
          normalize_latent_code=False,
          name="keypoint_mapping",
      )
  
      outputs = mapping_network(input_data)
  
      self.assertEqual(outputs.shape[1], output_dimension)
  
file_path: graphics/tensorflow_graphics/projects/gan/architectures_style_gan_test.py
imports: from absl.testing import parameterized
  import numpy
  import tensorflow
  from tensorflow_graphics.projects.gan import architectures_style_gan
label_desc: Use image node_id 4 for calling the ArchitecturesStyleGanTest obj's underlying member method code with example usage: obj.test_style_based_generator_mapping_outputs_shape(batch_size) without return types