docstr: Test automatic propagation of the query's mask.
docstr_tok: ["Test","automatic","propagation","of","the","query","'s","mask","."]
code: def test_query_mask_propagation(self):
      """Test automatic propagation of the query's mask."""
      layer = layers.MultiHeadAttention(num_heads=2, key_dim=2)
      self.assertTrue(layer.supports_masking)
      query = np.array(
          [[1, 2, 3, 0, 0], [3, 3, 1, 1, 2], [1, 0, 0, 0, 0]]
      )
      masked_query = layers.Embedding(4, 8, mask_zero=True)(query)
      value = np.random.normal(size=(3, 3, 8))
      output = layer(query=masked_query, value=value)
      self.assertAllClose(masked_query._keras_mask, output._keras_mask)
  
imports: import os
  import numpy
  import pytest
  from absl.testing import parameterized
  from keras import backend
  from keras import constraints
  from keras import initializers
  from keras import layers
  from keras import models
  from keras import saving
  from keras import testing
label_desc: Use image node_id 6 for calling the MultiHeadAttentionTest obj's underlying member method code with example usage: obj.test_query_mask_propagation() without return types