tf.sparse.mask