o wi@s4ddlmZddlZddlmZGdddeZdS)) annotationsN) OpRunUnaryNumc@seZdZdddZdS)SoftmaxNcCsF|p|j}||j|dd}t|}||j|dd}||jfS)N)axiskeepdims)rmaxnpexpsumastypedtype)selfXrtmpYrZ/home/ubuntu/sommelier/.venv/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