
    h                     B    S SK rS SKrS SKJr  S SKJr   " S S\5      rg)    N)Base)expectc                   x    \ 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r
g)
ReduceSumSquare   Nc                     / SQn [         R                  " S/[         R                  S9nSn[        R                  R                  SSS/S/US	9n[         R                  " SS
/SS//SS/SS//SS/SS///[         R                  S9n[         R                  " [         R                  " U5      [        U5      US:H  S9n[        UXA/U/SS9  [         R                  R                  S5        [         R                  R                  SSU 5      R                  [         R                  5      n[         R                  " [         R                  " U5      [        U5      US:H  S9n[        UXA/U/SS9  g )N      r      dtyper   r   dataaxesreducedinputsoutputskeepdimsr   r
                  	   
      r   axisr   .test_reduce_sum_square_do_not_keepdims_exampler   r   name-test_reduce_sum_square_do_not_keepdims_randomnparrayint64onnxhelper	make_nodefloat32sumsquaretupler   randomseeduniformastypeshaper   r   noder   r   s         e/var/www/fran/franai/venv/lib/python3.13/site-packages/onnx/backend/test/case/node/reducesumsquare.pyexport_do_not_keepdims&ReduceSumSquare.export_do_not_keepdims   sH   xx288,{{$$F#K	 % 
 xx!fq!fAA/1b'B81DERZZ
 &&4uT{XQR]S 	<IA		
 			qyy  b%077

C&&4uT{XQR]S<I@		
    c                     / SQn [         R                  " S/[         R                  S9nSn[        R                  R                  SSS/S/US9n[         R                  " SS	/S
S//SS/SS//SS/SS///[         R                  S9n[         R                  " [         R                  " U5      [        U5      US:H  S9n[        UXA/U/SS9  [         R                  R                  S5        [         R                  R                  SSU 5      R                  [         R                  5      n[         R                  " [         R                  " U5      [        U5      US:H  S9n[        UXA/U/SS9  g )Nr	   r   r   r   r   r   r   r   r   r
   r   r   r   r   r   r   r   r   r   r   'test_reduce_sum_square_keepdims_exampler!   r   r#   &test_reduce_sum_square_keepdims_randomr%   r4   s         r7   export_keepdimsReduceSumSquare.export_keepdims5   sH   xx288,{{$$F#K	 % 
 xx!fq!fAA/1b'B81DERZZ
 &&4uT{XQR]S 	<I:		
 			qyy  b%077

C&&4uT{XQR]S<I9		
r:   c                     / SQn [         R                  " / [         R                  S9nSn[        R                  R                  SSS/S/US9n[         R                  " SS	/S
S//SS/SS//SS/SS///[         R                  S9n[         R                  " [         R                  " U5      S US:H  S9n[        UXA/U/SS9  [         R                  R                  S5        [         R                  R                  SSU 5      R                  [         R                  5      n[         R                  " [         R                  " U5      S US:H  S9n[        UXA/U/SS9  g )Nr	   r   r   r   r   r   r   r   r   r
   r   r   r   r   r   r   r   r   r   r   4test_reduce_sum_square_default_axes_keepdims_exampler!   r   r#   3test_reduce_sum_square_default_axes_keepdims_random)r&   r'   r(   r)   r*   r+   r,   r-   r.   r   r0   r1   r2   r3   r4   s         r7   export_default_axes_keepdims,ReduceSumSquare.export_default_axes_keepdims]   s<   xx"((+{{$$F#K	 % 
 xx!fq!fAA/1b'B81DERZZ
 &&4th!mL 	<IG		
 			qyy  b%077

C&&4th!mL<IF		
r:   c                     / SQn [         R                  " S/[         R                  S9nSn[        R                  R                  SSS/S/US	9n[         R                  " SS
/SS//SS/SS//SS/SS///[         R                  S9n[         R                  " [         R                  " U5      [        U5      US:H  S9n[        UXA/U/SS9  [         R                  R                  S5        [         R                  R                  SSU 5      R                  [         R                  5      n[         R                  " [         R                  " U5      [        U5      US:H  S9n[        UXA/U/SS9  g )Nr	   r   r   r   r   r   r   r   r   r
   r   r   r   r   r   r   r   r   r   r   5test_reduce_sum_square_negative_axes_keepdims_exampler!   r   r#   4test_reduce_sum_square_negative_axes_keepdims_randomr%   r4   s         r7   export_negative_axes_keepdims-ReduceSumSquare.export_negative_axes_keepdims   sH   xxBHH-{{$$F#K	 % 
 xx!fq!fAA/1b'B81DERZZ
 &&4uT{XQR]S 	<IH		
 			qyy  b%077

C&&4uT{XQR]S<IG		
r:   c                     / SQn Sn/ SQn[         R                  R                  SSS/S/US9n[        R                  " / [        R
                  S	9R                  U 5      n[        R                  " S/[        R                  S	9n[        R                  " [        R                  " U[        R
                  S	95      n[        UXE/U/S
S9  g )N)r   r   r   r   )r   r   r   r   r   r   r   r   r    test_reduce_sum_square_empty_setr!   )
r)   r*   r+   r&   r'   r,   reshaper(   zerosr   )r5   r   reduced_shaper6   r   r   r   s          r7   export_empty_set ReduceSumSquare.export_empty_set   s    !{{$$F#K	 % 
 xx"**-55e<xx288,((288MDE<I3		
r:    )returnN)__name__
__module____qualname____firstlineno__staticmethodr8   r>   rC   rI   rP   __static_attributes__rR   r:   r7   r   r      sh    %
 %
N %
 %
N #
 #
J %
 %
N 
 
r:   r   )numpyr&   r)   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   rR   r:   r7   <module>r]      s!   
   , .u
d u
r:   