
    h               	           S SK rS SKrS SKJr  S SKJr  SS\R                  S\S\S\R                  4S jjr	 SS\R                  S\S\S\R                  4S	 jjr
 " S
 S\5      rg)    N)Base)expectdataaxiskeepdimsreturnc                     [         R                  " XS9nUS:X  a  [         R                  " X15      nUR                  [         R                  5      $ N)r      )npargmaxexpand_dimsastypeint64r   r   r   results       \/var/www/fran/franai/venv/lib/python3.13/site-packages/onnx/backend/test/case/node/argmax.pyargmax_use_numpyr      s8    YYt'F1}-==""    c                     [         R                  " X5      n [         R                  " XS9nU R                  U   U-
  S-
  nU(       a  [         R                  " X15      nUR                  [         R                  5      $ r
   )r   flipr   shaper   r   r   r   s       r   "argmax_use_numpy_select_last_indexr      s\     774DYYt'FZZ&*F-==""r   c                       \ rS rSr\SS j5       r\SS j5       r\SS j5       r\SS j5       r\SS j5       r	\SS j5       r
\SS	 j5       r\SS
 j5       rSrg)ArgMax   Nc                     [         R                  " SS/SS//[         R                  S9n SnSn[        R                  R                  SS/S	/XS
9n[        XUS9n[        UU /U/SS9  [         R                  R                  SS/ SQ5      R                  [         R                  5      n [        XUS9n[        X0/U/SS9  g )N   r      
   dtyper   r   r   r   inputsoutputsr   r   r   r   test_argmax_no_keepdims_exampler$   r%   namer   r      test_argmax_no_keepdims_randomr   arrayfloat32onnxhelper	make_noder   r   randomuniformr   r   r   r   noder   s        r   export_no_keepdimsArgMax.export_no_keepdims   s    xx!Q!R)<{{$$fXz % 
 "$HE6H2		
 yy  b)4;;BJJG!$HE&8X	
r   c                  ~   [         R                  " SS/SS//[         R                  S9n SnSn[        R                  R                  SS/S/XS	9n[        XUS
9n[        X0/U/SS9  [         R                  R                  SS/ SQ5      R                  [         R                  5      n [        XUS
9n[        X0/U/SS9  g )Nr   r   r   r    r!   r   r   r   r#   r&   test_argmax_keepdims_exampler(   r*   r+   test_argmax_keepdims_randomr.   r6   s        r   export_keepdimsArgMax.export_keepdims7   s    xx!Q!R)<{{$$fXz % 
 "$HE&8V	
 yy  b)4;;BJJG!$HE&8U	
r   c                  z   [         R                  " SS/SS//[         R                  S9n Sn[        R                  R                  SS/S/US	9n[        XS
9n[        UU /U/SS9  [         R                  R                  SS/ SQ5      R                  [         R                  5      n [        XS
9n[        UU /U/SS9  g )Nr   r   r   r    r!   r   r   r   )r$   r%   r   r    test_argmax_default_axis_exampler(   r*   r+   test_argmax_default_axis_randomr.   r   r   r7   r   s       r   export_default_axes_keepdims#ArgMax.export_default_axes_keepdimsL   s    xx!Q!R)<{{$$fXzH % 

 "$:6H3		
 yy  b)4;;BJJG!$:6H2		
r   c                     [         R                  " SS/SS//[         R                  S9n SnSn[        R                  R                  SS/S	/XS
9n[        XUS9n[        UU /U/SS9  [         R                  R                  SS/ SQ5      R                  [         R                  5      n [        XUS9n[        UU /U/SS9  g )Nr   r   r   r    r!   r   r   r   r#   r&   *test_argmax_negative_axis_keepdims_exampler(   r*   r+   )test_argmax_negative_axis_keepdims_randomr.   r6   s        r   export_negative_axis_keepdims$ArgMax.export_negative_axis_keepdimsg   s    xx!Q!R)<{{$$fXz % 
 "$HE6H=		
 yy  b)4;;BJJG!$HE6H<		
r   c            	         [         R                  " SS/SS//[         R                  S9n SnSn[        R                  R                  SS/S	/UUS
S9n[        XUS9n[        UU /U/SS9  [         R                  R                  SS/ SQ5      R                  [         R                  5      n [        XUS9n[        UU /U/SS9  g )Nr   r   r    r!   r   r   r   r   r   Tr$   r%   r   r   select_last_indexr&   1test_argmax_no_keepdims_example_select_last_indexr(   r*   r+   0test_argmax_no_keepdims_random_select_last_indexr   r/   r0   r1   r2   r3   r   r   r4   r5   r   r6   s        r   $export_no_keepdims_select_last_index+ArgMax.export_no_keepdims_select_last_index   s    xx!Q!R)<{{$$8J" % 
 4DhW6HD		
 yy  b)4;;BJJG3DhW6HC		
r   c            	         [         R                  " SS/SS//[         R                  S9n SnSn[        R                  R                  SS/S/UUS	S
9n[        XUS9n[        UU /U/SS9  [         R                  R                  SS/ SQ5      R                  [         R                  5      n [        XUS9n[        UU /U/SS9  g )Nr   r   r    r!   r   r   r   r   TrM   r&   .test_argmax_keepdims_example_select_last_indexr(   r*   r+   -test_argmax_keepdims_random_select_last_indexrQ   r6   s        r   !export_keepdims_select_last_index(ArgMax.export_keepdims_select_last_index   s    xx!Q!R)<{{$$8J" % 
 4DhW6HA		
 yy  b)4;;BJJG3DhW6H@		
r   c                  |   [         R                  " SS/SS//[         R                  S9n Sn[        R                  R                  SS/S/US	S
9n[        XS9n[        UU /U/SS9  [         R                  R                  SS/ SQ5      R                  [         R                  5      n [        XS9n[        UU /U/SS9  g )Nr   r   r    r!   r   r   r   r   T)r$   r%   r   rN   r@   2test_argmax_default_axis_example_select_last_indexr(   r*   r+   1test_argmax_default_axis_random_select_last_indexrQ   rC   s       r   .export_default_axes_keepdims_select_last_index5ArgMax.export_default_axes_keepdims_select_last_index   s    xx!Q!R)<{{$$8J" % 
 4DL6HE		
 yy  b)4;;BJJG3DL6HD		
r   c            	         [         R                  " SS/SS//[         R                  S9n SnSn[        R                  R                  SS/S	/UUS
S9n[        XUS9n[        UU /U/SS9  [         R                  R                  SS/ SQ5      R                  [         R                  5      n [        XUS9n[        UU /U/SS9  g )Nr   r   r    r!   rG   r   r   r   r   TrM   r&   <test_argmax_negative_axis_keepdims_example_select_last_indexr(   r*   r+   ;test_argmax_negative_axis_keepdims_random_select_last_indexrQ   r6   s        r   /export_negative_axis_keepdims_select_last_index6ArgMax.export_negative_axis_keepdims_select_last_index   s    xx!Q!R)<{{$$8J" % 
 4DhW6HO		
 yy  b)4;;BJJG3DhW6HN		
r    )r   N)__name__
__module____qualname____firstlineno__staticmethodr8   r=   rD   rJ   rR   rW   r\   ra   __static_attributes__rc   r   r   r   r      s    
 
. 
 
( 
 
4 
 
4 
 
> 
 
> 
 
< 
 
r   r   )r   r   )r   T)numpyr   r1   onnx.backend.test.case.baser   onnx.backend.test.case.noder   ndarrayintr   r   r   rc   r   r   <module>ro      s|   
   , .#2:: #S # #BJJ # 6:#
**##/2#ZZ#a
T a
r   