code: def cosine_similarity_vec(num_tokens, num_removed_vec):
      
      remaining = -np.array(num_removed_vec) + num_tokens
      return remaining / (np.sqrt(num_tokens + 1e-6) * np.sqrt(remaining + 1e-6))
docs: Return cosine similarity between a binary vector with all ones
      of length ``num_tokens`` and vectors of the same length with
      ``num_removed_vec`` elements set to zero.