
    Sic
                    \    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9n[         R                  " U 5      n[         R                  " U[         R                  " X!S9SUS9  U$ )Naxis   )npargmax
zeros_likeput_along_axisexpand_dims)xr   x_argmaxys       b/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/onnx/backend/test/case/node/hardmax.pyhardmaxr      s@    yy&H
aAa<adKH    c                  <    \ rS rSr\SS j5       r\SS j5       rSrg)Hardmax   c                    [         R                  R                  SS/S/S9n [        R                  " / SQ/ SQ/ SQ/ SQ/5      R                  [        R                  5      n[        U5      n[        X/U/S	S
9  [        R                  " / SQ/5      R                  [        R                  5      n[        U5      n[        X/U/SS
9  g )Nr   r   r   inputsoutputs)   r   r	      )r      r	   r   )r   r	   r   r   )r   r	   r   r   test_hardmax_exampler   r   name)r   r   r   r	   test_hardmax_one_hot)	onnxhelper	make_noder
   arrayastypefloat32r   r   )noder   r   s      r   exportHardmax.export   s    {{$$5E % 
 HHlL,MNUUJJ
 AJtC!3IJ HHl^$++BJJ7 AJtC!3IJr   c                    [         R                  R                  SSS5      R                  [         R                  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 )Nr      r   r   r   r   r   )r   r   r   r   test_hardmax_axis_0r    r	   test_hardmax_axis_1r   test_hardmax_axis_2test_hardmax_negative_axisr   test_hardmax_default_axis)
r
   randomrandnr'   r(   r#   r$   r%   r   r   )r   r)   r   s      r   export_hardmax_axisHardmax.export_hardmax_axis/   s   IIOOAq!$++BJJ7{{$$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*   r6   __static_attributes__r8   r   r   r   r      s,    K K2 ,P ,Pr   r   )r1   )r   
np.ndarrayr   intr9   rA   )
__future__r   numpyr
   r#   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r   r8   r   r   <module>rG      s+    #   , .HPd HPr   