Making the parameter num_dim_ tunnable for a kd_tree index