Ë
    b¤Igú   ã                  ór   — d dl mZ d dlZd dlZd dlmZ d dlmZ d	d
d„Z		 d	 	 	 	 	 	 	 d
d„Z
 G d„ de«      Zy)é    )ÚannotationsN)ÚBase)Úexpectc                ó¤   — t        j                  | |¬«      }|dk(  rt        j                  ||«      }|j                  t         j                  «      S ©N)Úaxisé   )ÚnpÚargminÚexpand_dimsÚastypeÚint64©Údatar   ÚkeepdimsÚresults       ú_/var/www/html/answerous/venv/lib/python3.12/site-packages/onnx/backend/test/case/node/argmin.pyÚargmin_use_numpyr      s<   € ÜY‰Yt $Ô'€FØ1‚}Ü—‘ ¨Ó-ˆØ=‰=œŸ™Ó"Ð"ó    c                óô   — t        j                  | |«      } t        j                  | |¬«      }| j                  |   |z
  dz
  }|rt        j                  ||«      }|j                  t         j                  «      S r   )r
   Úflipr   Úshaper   r   r   r   s       r   Ú"argmin_use_numpy_select_last_indexr      sb   € ô 7‰74˜Ó€DÜY‰Yt $Ô'€FØZ‰Z˜Ñ Ñ&¨Ñ*€FÙÜ—‘ ¨Ó-ˆØ=‰=œŸ™Ó"Ð"r   c                  óœ   — e Zd Zed
d„«       Zed
d„«       Zed
d„«       Zed
d„«       Zed
d„«       Zed
d„«       Z	ed
d„«       Z
ed
d„«       Zy	)ÚArgMinc                 óª  — t        j                  ddgddggt         j                  ¬«      } d}d}t        j                  j                  ddgd	g||¬
«      }t        | ||¬«      }t        || g|gd¬«       t         j                  j                  ddg d¢«      j                  t         j                  «      } t        | ||¬«      }t        || g|gd¬«       y )Né   r	   é   é
   ©Údtyper   r   r   r   ©ÚinputsÚoutputsr   r   ©r   r   Útest_argmin_no_keepdims_example©r#   r$   Únameéöÿÿÿ©r   r   é   Útest_argmin_no_keepdims_random©r
   ÚarrayÚfloat32ÚonnxÚhelperÚ	make_noder   r   ÚrandomÚuniformr   ©r   r   r   Únoder   s        r   Úexport_no_keepdimszArgMin.export_no_keepdims    sÊ   € äx‰x˜!˜Q˜ ! R Ð)´·±Ô<ˆØˆØˆÜ{‰{×$Ñ$Ø˜f˜X°¨zÀÈxð %ó 
ˆô " $¨T¸HÔEˆÜØØ6ØHØ2õ		
ô y‰y× Ñ   bª)Ó4×;Ñ;¼B¿J¹JÓGˆä! $¨T¸HÔEˆÜØ˜$˜¨&¨Ð8Xö	
r   c                 óª  — t        j                  ddgddggt         j                  ¬«      } d}d}t        j                  j                  ddgdg||¬	«      }t        | ||¬
«      }t        || g|gd¬«       t         j                  j                  ddg d¢«      j                  t         j                  «      } t        | ||¬
«      }t        || g|gd¬«       y )Nr   r	   r   r   r    r   r   r   r"   r%   Útest_argmin_keepdims_exampler'   r)   r*   Útest_argmin_keepdims_randomr-   r5   s        r   Úexport_keepdimszArgMin.export_keepdims8   sÈ   € äx‰x˜!˜Q˜ ! R Ð)´·±Ô<ˆØˆØˆÜ{‰{×$Ñ$Ø˜f˜X°¨zÀÈxð %ó 
ˆô " $¨T¸HÔEˆÜØ˜$˜¨&¨Ð8Võ	
ô y‰y× Ñ   bª)Ó4×;Ñ;¼B¿J¹JÓGˆä! $¨T¸HÔEˆÜØ˜$˜¨&¨Ð8Uö	
r   c                 ó   — t        j                  ddgddggt         j                  ¬«      } d}t        j                  j                  ddgdg|¬	«      }t        | |¬
«      }t        || g|gd¬«       t         j                  j                  ddg d¢«      j                  t         j                  «      } t        | |¬
«      }t        || g|gd¬«       y )Nr   r	   r   r   r    r   r   r   )r#   r$   r   ©r   Ú test_argmin_default_axis_exampler'   r)   r*   Útest_argmin_default_axis_randomr-   ©r   r   r6   r   s       r   Úexport_default_axes_keepdimsz#ArgMin.export_default_axes_keepdimsM   sÁ   € äx‰x˜!˜Q˜ ! R Ð)´·±Ô<ˆØˆÜ{‰{×$Ñ$Ø˜f˜X°¨zÀHð %ó 
ˆô
 " $°Ô:ˆÜØØ6ØHØ3õ		
ô y‰y× Ñ   bª)Ó4×;Ñ;¼B¿J¹JÓGˆä! $°Ô:ˆÜØØ6ØHØ2ö		
r   c                 óª  — t        j                  ddgddggt         j                  ¬«      } d}d}t        j                  j                  ddgd	g||¬
«      }t        | ||¬«      }t        || g|gd¬«       t         j                  j                  ddg d¢«      j                  t         j                  «      } t        | ||¬«      }t        || g|gd¬«       y )Nr   r	   r   r   r    éÿÿÿÿr   r   r   r"   r%   Ú*test_argmin_negative_axis_keepdims_exampler'   r)   r*   Ú)test_argmin_negative_axis_keepdims_randomr-   r5   s        r   Úexport_negative_axis_keepdimsz$ArgMin.export_negative_axis_keepdimsh   sÌ   € äx‰x˜!˜Q˜ ! R Ð)´·±Ô<ˆØˆØˆÜ{‰{×$Ñ$Ø˜f˜X°¨zÀÈxð %ó 
ˆô " $¨T¸HÔEˆÜØØ6ØHØ=õ		
ô y‰y× Ñ   bª)Ó4×;Ñ;¼B¿J¹JÓGˆä! $¨T¸HÔEˆÜØØ6ØHØ<ö		
r   c                 ó¬  — t        j                  ddgddggt         j                  ¬«      } d}d}t        j                  j                  ddgd	g||d
¬«      }t        | ||¬«      }t        || g|gd¬«       t         j                  j                  ddg d¢«      j                  t         j                  «      } t        | ||¬«      }t        || g|gd¬«       y )Nr   r   r   r    r	   r   r   r   r   T©r#   r$   r   r   Úselect_last_indexr%   Ú1test_argmin_no_keepdims_example_select_last_indexr'   r)   r*   Ú0test_argmin_no_keepdims_random_select_last_index©r
   r.   r/   r0   r1   r2   r   r   r3   r4   r   r5   s        r   Ú$export_no_keepdims_select_last_indexz+ArgMin.export_no_keepdims_select_last_indexƒ   sÓ   € äx‰x˜!˜Q˜ ! R Ð)´·±Ô<ˆØˆØˆÜ{‰{×$Ñ$ØØ8ØJØØØ"ð %ó 
ˆô 4°D¸tÈhÔWˆÜØØ6ØHØDõ		
ô y‰y× Ñ   bª)Ó4×;Ñ;¼B¿J¹JÓGˆä3°D¸tÈhÔWˆÜØØ6ØHØCö		
r   c                 ó¬  — t        j                  ddgddggt         j                  ¬«      } d}d}t        j                  j                  ddgdg||d	¬
«      }t        | ||¬«      }t        || g|gd¬«       t         j                  j                  ddg d¢«      j                  t         j                  «      } t        | ||¬«      }t        || g|gd¬«       y )Nr   r   r   r    r	   r   r   r   TrH   r%   Ú.test_argmin_keepdims_example_select_last_indexr'   r)   r*   Ú-test_argmin_keepdims_random_select_last_indexrL   r5   s        r   Ú!export_keepdims_select_last_indexz(ArgMin.export_keepdims_select_last_index£   sÓ   € äx‰x˜!˜Q˜ ! R Ð)´·±Ô<ˆØˆØˆÜ{‰{×$Ñ$ØØ8ØJØØØ"ð %ó 
ˆô 4°D¸tÈhÔWˆÜØØ6ØHØAõ		
ô y‰y× Ñ   bª)Ó4×;Ñ;¼B¿J¹JÓGˆä3°D¸tÈhÔWˆÜØØ6ØHØ@ö		
r   c                 ó¢  — t        j                  ddgddggt         j                  ¬«      } d}t        j                  j                  ddgdg|d	¬
«      }t        | |¬«      }t        || g|gd¬«       t         j                  j                  ddg d¢«      j                  t         j                  «      } t        | |¬«      }t        || g|gd¬«       y )Nr   r   r   r    r	   r   r   r   T)r#   r$   r   rI   r=   Ú2test_argmin_default_axis_example_select_last_indexr'   r)   r*   Ú1test_argmin_default_axis_random_select_last_indexrL   r@   s       r   Ú.export_default_axes_keepdims_select_last_indexz5ArgMin.export_default_axes_keepdims_select_last_indexÃ   sÇ   € äx‰x˜!˜Q˜ ! R Ð)´·±Ô<ˆØˆÜ{‰{×$Ñ$ØØ8ØJØØ"ð %ó 
ˆô 4°DÀ8ÔLˆÜØØ6ØHØEõ		
ô y‰y× Ñ   bª)Ó4×;Ñ;¼B¿J¹JÓGˆä3°DÀ8ÔLˆÜØØ6ØHØDö		
r   c                 ó¬  — t        j                  ddgddggt         j                  ¬«      } d}d}t        j                  j                  ddgd	g||d
¬«      }t        | ||¬«      }t        || g|gd¬«       t         j                  j                  ddg d¢«      j                  t         j                  «      } t        | ||¬«      }t        || g|gd¬«       y )Nr   r   r   r    rC   r	   r   r   r   TrH   r%   Ú<test_argmin_negative_axis_keepdims_example_select_last_indexr'   r)   r*   Ú;test_argmin_negative_axis_keepdims_random_select_last_indexrL   r5   s        r   Ú/export_negative_axis_keepdims_select_last_indexz6ArgMin.export_negative_axis_keepdims_select_last_indexâ   sÓ   € äx‰x˜!˜Q˜ ! R Ð)´·±Ô<ˆØˆØˆÜ{‰{×$Ñ$ØØ8ØJØØØ"ð %ó 
ˆô 4°D¸tÈhÔWˆÜØØ6ØHØOõ		
ô y‰y× Ñ   bª)Ó4×;Ñ;¼B¿J¹JÓGˆä3°D¸tÈhÔWˆÜØØ6ØHØNö		
r   N)ÚreturnÚNone)Ú__name__Ú
__module__Ú__qualname__Ústaticmethodr7   r;   rA   rF   rM   rQ   rU   rY   © r   r   r   r      s    „ Øò
ó ð
ð. ò
ó ð
ð( ò
ó ð
ð4 ò
ó ð
ð4 ò
ó ð
ð> ò
ó ð
ð> ò
ó ð
ð< ò
ó ñ
r   r   )r   r	   )r   ú
np.ndarrayr   Úintr   rb   rZ   ra   )r   T)Ú
__future__r   Únumpyr
   r0   Úonnx.backend.test.case.baser   Úonnx.backend.test.case.noder   r   r   r   r`   r   r   ú<module>rg      sS   ðõ #ã ã Ý ,Ý .ô#ð 6:ð#Ø
ð#Øð#Ø/2ð#àó#ôa
ˆTõ a
r   