tf.sparse.retain