U qh@s<ddlmZddlZddlmZddZGdddeZdS)) annotationsN)OpRuncCs.tj|tjdtj|tjdtj|tjdfS)NZdtype)nparrayint64)indicesinverse_indicescountsr @/tmp/pip-unpacked-wheel-xnis5xre/onnx/reference/ops/op_unique.py_specify_int64 sr c@seZdZdddZdS)UniqueNc s|dkst|r,t|ddd\}}}}ntj|ddd|d\}}}}t|jjdkr^|fS|st|}tt|t t|||}tj ||dd}tj fdd|Dtj d}||}t |||\}}}|}t|jjdkr||fSt|jjd kr|||fS||||fS) NT)axisrcsg|] }|qSr r ).0iZinverse_indices_mapr r &szUnique._run..r)risnanuniquelenZ onnx_nodeoutputZargsortdictzipZarangeZtakeZasarrayrr Zsqueeze) selfxrsortedyrr r Zargsorted_indicesr rr _runsD   z Unique._run)NN)__name__ __module__ __qualname__r!r r r r rsr) __future__rZnumpyrZonnx.reference.op_runrr rr r r r s