code: def test_attention_correctness(self):
      query = np.array([[[1.0, 0.0], [0.0, 1.0]]])
      key = np.array([[[0.0, 1.0], [1.0, 0.0]]])
      value = np.array([[[1.0, 2.0], [3.0, 4.0]]])
  
      layer = layers.AdditiveAttention(use_scale=False)
      output, scores = layer(
          [query, value, key],
          return_attention_scores=True,
      )
      self.assertAllClose(
          output, [[[1.727, 2.727], [2.272, 3.272]]], atol=1e-3
      )
      self.assertAllClose(
          scores, [[[0.636, 0.363], [0.363, 0.636]]], atol=1e-3
      )
  
file_path: keras/keras/layers/attention/additive_attention_test.py
imports: import numpy
  from keras import layers
  from keras import testing
label_desc: Use image node_id 2 for calling the AdditiveAttentionTest obj's underlying member method code with example usage: obj.test_attention_correctness() without return types