2Y d dl mZ ddlmZ d dlmZ d dlZ G d de Z G d d e Z G d
d e Z
ddZ ddZ ddZ
ddZdS ) ) annotations )Array)
NamedTupleNc 8 e Zd ZU ded< ded< ded< ded< dS )UniqueAllResultr valuesindicesinverse_indicescountsN__name__
__module__ __qualname____annotations__ r/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/array_api/_set_functions.pyr r s7 MMMNNNMMMMMr r c $ e Zd ZU ded< ded< dS )UniqueCountsResultr r r Nr
r r r r r s" MMMMMMMMr r c $ e Zd ZU ded< ded< dS )UniqueInverseResultr r r Nr
r r r r r ! s' MMMr r xr returnc 0 t j | j dddd \ }}}}| | 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_arrayreshapeshaper r _new)r r r
r r s r
unique_allr( & s 02y 0 0 0,FG_f &--ag66O
6
7
?##
6 r c d t j | j dddd }t d |D S )NTFr c 6 g | ]}t j | S r )r r' ).0is r
z!unique_counts..G s ; ; ;!
1
; ; ;r )r" r# r$ r r ress r
unique_countsr0 > sE
) C ; ;s ; ; ; <r9 s# " " " " " " j
*
0 = = = =P P P P&
r