o toi4@s4ddlmZddlZddlmZGdddeZdS)) annotationsN) OpRunUnaryNumc@seZdZdddZdS)HardmaxNcCsT|p|j}|jdkr |fStj||d}t|}tj|tj||dd|d|fS)Nr)axis)rsizenpargmax zeros_likeput_along_axis expand_dims)selfxrx_argmaxyrQ/home/ubuntu/.local/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