U rh @stddlmZddlZddlZddlmZddlmZddddddd d Z ddddddd d Z GdddeZ dS)) annotationsN)Base)expectz np.ndarrayint)dataaxiskeepdimsreturncCs.tj||d}|dkr"t||}|tjSN)rr)npargmax expand_dimsastypeint64rrr resultrF/tmp/pip-unpacked-wheel-xnis5xre/onnx/backend/test/case/node/argmax.pyargmax_use_numpy s rTcCsHt||}tj||d}|j||d}|rs