gTddlZddlZdZdZdZdZdZd dZdZ d Z d Z d Z dS) NcZtj|}||}tj|d}dx}} ||}|dkr-tjtj|||j| |}|j tj j krtj||j fd}n5|j }|j}||zdz |z} tj| ||z|fd}|dkr-tjtj|||j||||||||n6#||||||||wwxYw||fS)zy returns the inverted lists content as a pair of (list_ids, list_codes). The codes are reshaped to a proper size int64dtypeNruint8)faissdowncast_InvertedLists list_sizenpzerosget_idsmemcpyswig_ptrnbytes get_codes code_size InvertedListsINVALID_CODE_SIZE n_per_block block_size release_ids release_codes) invlistsllslist_idsidscodes list_codesnpbbsls_rounds Y/var/www/html/mpstechhub/venv/lib/python3.11/site-packages/faiss/contrib/inspect_tools.py get_invlistr% s+H55H   A  Bx'***HC%-q!! 66 L113 H H H""1%%  !4!F F F2x'9":'JJJJJ&C$BS1 ,H8R3Y"z%get_invlist_sizes..+s7  1rr)r arrayrangenlist)rs`r$get_invlist_sizesr0)sQ 8x~&&   r,c h|jjD]$}t|dt||%dS)z3 list values all fields of an object known to SWIG z = N) __class____swig_getmethods__printgetattr)objnames r$print_object_fieldsr81sK 100 ..'#t,,..////00r,ctj|j}||j|j|jS)z% return the PQ centroids as an array )r vector_to_array centroidsreshapeMksubdsub)pqcens r$get_pq_centroidsrB8s0   - -C ;;rtRWbg . ..r,ctj|j}tj|j|j|j}||fS)zo extract matrix + bias from the PCA object works for any linear transform (OPQ, random rotation, etc.) )r r:bAr<d_outd_in)pcarDrEs r$get_LinearTransform_matrixrI>sE ce$$A ce$$,,SYAAA a4Kr,c0|j\}}||j|fksJtj|||du}tj||j|tj||jd|_||S)zA make a linear transform from a matrix and a bias term (optional)NT) shaper LinearTransformcopy_array_to_vectorravelrErD is_trainedset_is_orthonormal)rErDrFrGlts r$make_LinearTransform_matrixrRGs'KE4}w5)####  tUATM : :B qwwyy"$///} "1bd+++BM Ir,ctj|jd|jtj|jfdt |jDS)z. return to codebooks of an additive quantizer cBg|]}||dzS)rr()r)r*co codebookss r$r+z4get_additive_quantizer_codebooks..Ys>     "Q%1q5 /"   r,)r r:rWr<dcodebook_offsetsr.r=)aqrVrWs @@r$ get_additive_quantizer_codebooksr[Uss%bl33;;BEEI  r2 3 3B     rt   r,ctj|jd}||j|jS)z1 copy and return the data matrix in an IndexFlat float32)r r:rviewr<ntotalrX)indexxbs r$ get_flat_datarb_s:  u{ + + 0 0 ; ;B ::elEG , ,,r,cptj|j|j|jS)z2 get the codes from an indexFlatCodes as an array )r r:rr<r_r) index_flats r$get_flat_codesrees3  !1 2 2 : ::/ 1 11r,c|}tj|j|jfd}t jt j||j|j |S)zf get the neighbor list for the vectors stored in the NSG structure, as a N-by-K matrix of indices int32r) get_final_graphr r NKr rrdatar)nsggraph neighborss r$get_NSG_neighborsroksf    ! !E%'57+7;;;I L y!!  r,)N) numpyr r r%r0r8rBrIrRr[rbreror(r,r$rqs     >000///     --- 111     r,