code: def test_initializer(self):
      # Test with a specified initializer.
      layer = layers.GroupedQueryAttention(
          num_query_heads=16,
          num_key_value_heads=16,
          head_dim=64,
          kernel_initializer=initializers.TruncatedNormal(stddev=0.02),
      )
      layer.build((2, 4, 8), (2, 4, 8))
  
      # Make sure the sub layers have different kernel init value.
      self.assertNotAllClose(
          layer._query_dense.kernel,
          layer._key_dense.kernel,
      )
      self.assertNotAllClose(
          layer._query_dense.kernel,
          layer._value_dense.kernel,
      )
      self.assertNotAllClose(
          layer._query_dense.kernel,
          layer._output_dense.kernel,
      )
  
file_path: keras/keras/layers/attention/grouped_query_attention_test.py
imports: import numpy
  import pytest
  from absl.testing import parameterized
  from keras import backend
  from keras import initializers
  from keras import layers
  from keras import testing
label_desc: Use image node_id 4 for calling the GroupedQueryAttentionTest obj's underlying member method code with example usage: obj.test_initializer() without return types