tf.ragged.boolean_mask