
    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X-
  [         R                  " U5      -
  $ )NT)r   keepdims)npmaxexpsumlog)r   r   x_maxtmpss        `/var/www/fran/franai/venv/lib/python3.13/site-packages/onnx/backend/test/case/node/logsoftmax.py
logsoftmaxr      sG    FF1$/E
&&
C
s-AI""    c                   <    \ rS rSr\SS j5       r\SS j5       rSrg)
LogSoftmax   Nc                      [         R                  R                  SS/S/S9n [        R                  " / SQ/5      R                  [        R                  5      n[        U5      n[        X/U/SS9  g )Nr   r   yinputsoutputs)r      test_logsoftmax_example_1r   r   name)	onnxhelper	make_noder
   arrayastypefloat32r   r   )noder   r   s      r   exportLogSoftmax.export   se    {{$$5E % 

 HHj\"))"**5 qMtC!3NOr   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_logsoftmax_large_numberr    r-         r   )r   r   r   )r   test_logsoftmax_axis_0r   test_logsoftmax_axis_1r,   test_logsoftmax_axis_2r   test_logsoftmax_negative_axistest_logsoftmax_default_axis)r
   r%   r&   r'   r   r"   r#   r$   r   absrandomrandn)r   r   r(   s      r   export_logsoftmax_axis!LogSoftmax.export_logsoftmax_axis!   s   HHl$@ABII"**U qM{{$$5E % 

 	tC!3QRFF299??1a+222::>?{{$$5E	 % 
 qq!tC!3KL{{$$5E	 % 
 qq!tC!3KL{{$$5E	 % 
 qq!tC!3KL{{$$5E	 % 
 qr"tC!3RS {{$$5E % 

 	tC!3QRr    )r   N)__name__
__module____qualname____firstlineno__staticmethodr)   r9   __static_attributes__r;   r   r   r   r      s,    
P 
P 9S 9Sr   r   )r   )numpyr
   r"   onnx.backend.test.case.baser   onnx.backend.test.case.noder   ndarrayintr   r   r;   r   r   <module>rG      sE   
   , .#"** #C # #HS HSr   