tf.compat.v1.boolean_mask