code: def test_generator_output_size(
      self, batch_size, normalize_latent_code
  ):
      input_data = np.ones(shape=(batch_size, 8), dtype=np.float32)
      (
          generator,
          _,
          _,
      ) = architectures_style_gan_v2.create_style_based_generator(
          latent_code_dimension=8,
          upsampling_blocks_num_channels=(8, 8),
          normalize_latent_code=normalize_latent_code,
      )
      expected_size = 16
  
      output = generator(input_data)
  
      self.assertSequenceEqual(
          output.shape, (batch_size, expected_size, expected_size, 3)
      )
  
file_path: graphics/tensorflow_graphics/projects/gan/architectures_style_gan_v2_test.py
imports: from absl.testing import parameterized
  import numpy
  import tensorflow
  from tensorflow_graphics.projects.gan import architectures_style_gan_v2
label_desc: Use image node_id 1 for calling the ArchitecturesStyleGanV2Test obj's underlying member method code with example usage: obj.test_generator_output_size(batch_size, normalize_latent_code) without return types