code: def test_call(self):
      list_mask = tf.constant([[1, 1], [1, 0]], dtype=tf.int32)
      self_attention_mask = tf.constant(
          [[[1, 1], [1, 1]], [[1, 0], [1, 0]]], dtype=tf.int32
      )
      self.assertAllClose(
          layers.SelfAttentionMask()(inputs=[list_mask, list_mask]),
          self_attention_mask,
      )
  
imports: import tensorflow
  from tensorflow_ranking.python.keras import layers
label_desc: Use image node_id 2 for calling the SelfAttentionMaskTest obj's underlying member method code with example usage: obj.test_call() without return types