o toi@s<ddlmZddlZddlmZddZGdddeZdS)) annotationsN)OpRuncCs.tj|tjdtj|tjdtj|tjdfS)Ndtype)nparrayint64)indicesinverse_indicescountsr P/home/ubuntu/.local/lib/python3.10/site-packages/onnx/reference/ops/op_unique.py_specify_int64 src@seZdZdddZdS)UniqueNc s |dus t|rt|ddd\}}}}ntj|ddd|d\}}}}t|jjdkr/|fS|sct|}tt|t t|dd||}tj ||dd}tj fdd|Dtj d}||}t |||\}}}|d }t|jjd kr}||fSt|jjd kr|||fS||||fS) NT)axis)strictrcsg|]}|qSr r ).0iinverse_indices_mapr r &szUnique._run..r)risnanuniquelen onnx_nodeoutputargsortdictziparangetakeasarrayrrreshape) selfxrsortedyr r r argsorted_indicesr rr _runs6     z Unique._run)NN)__name__ __module__ __qualname__r,r r r r rsr) __future__rnumpyronnx.reference.op_runrrrr r r r s