o toi@s4ddlmZddlZddlmZGdddeZdS)) annotationsN) OpRunUnaryNumc@seZdZdddZdS)SoftmaxNcCsV|jdkr|fS|p |j}||j|dd}t|}||j|dd}||jfS)Nr)axiskeepdims)sizermaxnpexpsumastypedtype)selfXrtmpYrQ/home/ubuntu/.local/lib/python3.10/site-packages/onnx/reference/ops/op_softmax.py_run s   z Softmax._run)N)__name__ __module__ __qualname__rrrrrr sr) __future__rnumpyr onnx.reference.ops._oprrrrrrs