
    h[
                         S SK rS SKrS SKJr  S SKJr  S
S\R                  S\S\R                  4S jjr	 " S S	\5      r
g)    N)Base)expectxaxisreturnc                     [         R                  " XSS9n[         R                  " X-
  5      n[         R                  " X1SS9nX4-  $ )NT)r   keepdims)npmaxexpsum)r   r   x_maxtmpss        ]/var/www/fran/franai/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   Nc                      [         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    )r   N)__name__
__module____qualname____firstlineno__staticmethodr)   r9   __static_attributes__r;   r   r   r   r      s,    	K 	K 9P 9Pr   r   )r   )numpyr
   r"   onnx.backend.test.case.baser   onnx.backend.test.case.noder   ndarrayintr   r   r;   r   r   <module>rG      sE   
   , .rzz  bjj GPd GPr   