
    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9n[         R                  " U 5      n[         R                  " U[         R                  " X!S9SUS9  U$ )Nr      )npargmax
zeros_likeput_along_axisexpand_dims)r   r   x_argmaxys       ]/var/www/fran/franai/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   Nc                     [         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    )r   N)__name__
__module____qualname____firstlineno__staticmethodr*   r6   __static_attributes__r8   r   r   r   r      s,    K K2 ,P ,Pr   r   )r1   )numpyr   r#   onnx.backend.test.case.baser   onnx.backend.test.case.noder   ndarrayintr   r   r8   r   r   <module>rD      sE   
   , .rzz  bjj HPd HPr   