
    Si~
                    \    S SK Jr  S SKrS SKrS SKJr  S SKJr  SS	S jjr	 " S S\5      r
g)
    )annotationsN)Base)expectc                    [         R                  " XSS9n[         R                  " X-
  5      n[         R                  " X1SS9nX4-  $ )NT)axiskeepdims)npmaxexpsum)xr   x_maxtmpss        b/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/onnx/backend/test/case/node/softmax.pysoftmaxr      s9    FF1$/E
&&
C
s-A7N    c                  <    \ rS rSr\SS j5       r\SS j5       rSrg)Softmax   c                     [         R                  R                  SS/S/S9n [        R                  " / SQ/5      R                  [        R                  5      n[        USS9n[        X/U/SS	9  g )
Nr   r   yinputsoutputs)r      r   r   test_softmax_exampler   r   name)	onnxhelper	make_noder	   arrayastypefloat32r   r   )noder   r   s      r   exportSoftmax.export   sf    {{$$5E % 

 HHj\"))"**5AAtC!3IJr   c                    [         R                  " / SQ/ SQ/5      R                  [         R                  5      n [	        U 5      n[
        R                  R                  SS/S/S9n[        X /U/SS9  [         R                  " [         R                  R                  S	S
S5      R                  [         R                  5      5      n [
        R                  R                  SS/S/SS9n[	        U SS9n[        X /U/SS9  [
        R                  R                  SS/S/SS9n[	        U SS9n[        X /U/SS9  [
        R                  R                  SS/S/SS9n[	        U SS9n[        X /U/SS9  [
        R                  R                  SS/S/SS9n[	        U SS9n[        X /U/SS9  [
        R                  R                  SS/S/S9n[        X /U/SS9  g )N)r   r         )i'  i'  i'  i'  r   r   r   r   test_softmax_large_numberr    r-         r   )r   r   r   r   test_softmax_axis_0r   test_softmax_axis_1r,   test_softmax_axis_2r   test_softmax_negative_axistest_softmax_default_axis)r	   r%   r&   r'   r   r"   r#   r$   r   absrandomrandn)r   r   r(   s      r   export_softmax_axisSoftmax.export_softmax_axis!   s   HHl$@ABII"**U AJ{{$$5E % 

 	tC!3NOFF299??1a+222::>?{{$$5E	 % 
 AAtC!3HI{{$$5E	 % 
 AAtC!3HI{{$$5E	 % 
 AAtC!3HI{{$$5E	 % 
 ABtC!3OP {{$$5E % 

 	tC!3NOr    N)returnNone)__name__
__module____qualname____firstlineno__staticmethodr)   r9   __static_attributes__r;   r   r   r   r      s,    	K 	K 9P 9Pr   r   )r   )r   
np.ndarrayr   intr<   rD   )
__future__r   numpyr	   r"   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r   r;   r   r   <module>rJ      s+    #   , .GPd GPr   