bg ddlmZddlmZddlmZddlZGddeZGdd eZ Gd d eZ ddZ ddZ ddZ ddZdS)) annotations)Array) NamedTupleNc8eZdZUded<ded<ded<ded<dS)UniqueAllResultrvaluesindicesinverse_indicescountsN__name__ __module__ __qualname____annotations__q/builddir/build/BUILD/cloudlinux-venv-1.0.7/venv/lib64/python3.11/site-packages/numpy/array_api/_set_functions.pyrrs7MMMNNNMMMMMrrc$eZdZUded<ded<dS)UniqueCountsResultrr r Nr rrrrrs"MMMMMMMMrrc$eZdZUded<ded<dS)UniqueInverseResultrr r Nr rrrrr!s'MMMrrxrreturnc0tj|jdddd\}}}}||j}t t j|t j|t j|t j|S)| Array API compatible wrapper for :py:func:`np.unique `. See its docstring for more information. TF return_counts return_indexreturn_inverse equal_nan)npunique_arrayreshapeshaperr_new)rr r r r s r unique_allr(&s 02y  000,FG_f&--ag66O  6 7 ?## 6   rcdtj|jdddd}td|DS)NTFrc6g|]}tj|Sr)rr').0is r z!unique_counts..Gs ;;;! 1 ;;;r)r"r#r$rrress r unique_countsr0>sE )     C ;;s;;; <r9s#"""""" j * 0 = = = =PPPP&      r