a ^g@sfddlZddlZddZddZddZdd Zd d Zdd d ZddZ ddZ ddZ ddZ dS)Nc Cs6t|}||}tj|dd}d}}z||}|dkrTtt|||j| |}|j tj j krtj||j fdd}n4|j }|j}||d|} tj| |||fdd}|dkrtt|||jW|dur||||dur|||n.|dur||||dur,|||0||fS)zy returns the inverted lists content as a pair of (list_ids, list_codes). The codes are reshaped to a proper size int64ZdtypeNrZuint8)faissZdowncast_InvertedLists list_sizenpzerosZget_idsmemcpyswig_ptrnbytes get_codes code_sizeZ InvertedListsZINVALID_CODE_SIZEZ n_per_block block_sizeZ release_idsZ release_codes) invlistslZlsZlist_idsZidscodesZ list_codesZnpbbsZls_roundri/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/faiss/contrib/inspect_tools.py get_invlist s4        rcs"tjfddtjDddS)z1 return the array of sizes of the inverted lists csg|]}|qSr)r.0irrr +sz%get_invlist_sizes..rr)rarrayrangeZnlistrrrrget_invlist_sizes)s rcCs*|jjD]}t|dt||qdS)z3 list values all fields of an object known to SWIG z = N) __class__Z__swig_getmethods__printgetattr)objnamerrrprint_object_fields1s r#cCs t|j}||j|j|jS)z% return the PQ centroids as an array )rvector_to_arrayZ centroidsreshapeMZksubZdsub)ZpqZcenrrrget_pq_centroids8s r'cCs,t|j}t|j|j|j}||fS)zo extract matrix + bias from the PCA object works for any linear transform (OPQ, random rotation, etc.) )rr$bAr%d_outd_in)Zpcar(r)rrrget_LinearTransform_matrix>s r,cCsn|j\}}|dur"|j|fks"Jt|||du}t||j|dur\t||jd|_||S)zA make a linear transform from a matrix and a bias term (optional)NT) shaperZLinearTransformZcopy_array_to_vectorZravelr)r(Z is_trainedZset_is_orthonormal)r)r(r*r+ltrrrmake_LinearTransform_matrixGs r/cs<t|jd|jt|jfddt|jDS)z. return to codebooks of an additive quantizer cs$g|]}||dqS)rrrco codebooksrrrYsz4get_additive_quantizer_codebooks..)rr$r3r%dZcodebook_offsetsrr&)Zaqrr1r get_additive_quantizer_codebooksUs   r5cCs"t|jd}||j|jS)z1 copy and return the data matrix in an IndexFlat Zfloat32)rr$rviewr%ntotalr4)indexxbrrr get_flat_data_sr:cCst|j|j|jS)z2 get the codes from an indexFlatCodes as an array )rr$rr%r7r )Z index_flatrrrget_flat_codeses r;cCs:|}tj|j|jfdd}tt||j|j |S)zf get the neighbor list for the vectors stored in the NSG structure, as a N-by-K matrix of indices Zint32r) Zget_final_graphrrNKrr r datar )ZnsggraphZ neighborsrrrget_NSG_neighborsksr@)N) numpyrrrrr#r'r,r/r5r:r;r@rrrrs