import tensorflow as tf

    masks1 = tf.constant([[0, 0, 0, 0, 0],
                          [1, 1, 1, 1, 1],
                          [1, 0, 1, 0, 1],
                          [0, 1, 0, 1, 0]], dtype=tf.int32)
    masks2 = tf.constant([[0, 0, 0, 0, 0],
                          [1, 1, 1, 1, 1],
                          [1, 1, 1, 1, 0],
                          [1, 0, 1, 1, 1]], dtype=tf.int32)
