o wiD@s4ddlmZddlZddlmZGdddeZdS)) annotationsN) OpRunUnaryNumc@seZdZdddZdS)HardmaxNcCsD|p|j}tj||d}t|}tj|tj||dd|d|fS)N)axis)rnpargmax zeros_likeput_along_axis expand_dims)selfxrx_argmaxyrZ/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/onnx/reference/ops/op_hardmax.py_run s   z Hardmax._run)N)__name__ __module__ __qualname__rrrrrr sr) __future__rnumpyronnx.reference.ops._oprrrrrrs