
    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                   d    \ 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r	g)	
ReduceMean   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                  " U[        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                  " U[        U5      US:H  S9n[        UXA/U/SS9  g )N      r      dtyper   r   dataaxesreducedinputsoutputskeepdims      r      (   7   <   axisr   (test_reduce_mean_do_not_keepdims_exampler   r   name
   'test_reduce_mean_do_not_keepdims_randomnparrayint64onnxhelper	make_nodefloat32meantupler   randomseeduniformastypeshaper   r   noder   r   s         `/var/www/fran/franai/venv/lib/python3.13/site-packages/onnx/backend/test/case/node/reducemean.pyexport_do_not_keepdims!ReduceMean.export_do_not_keepdims   s7   xx288,{{$$F#K	 % 
 xx!fr1g"a2q' 2b!Wr1g4FG**
 ''$U4[8q=I 	<I;		
 			qyy  b%077

C''$U4[8q=I<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                  " U[        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                  " U[        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   !test_reduce_mean_keepdims_exampler   r   r!   r"    test_reduce_mean_keepdims_randomr$   r2   s         r5   export_keepdimsReduceMean.export_keepdims6   s7   xx288,{{$$F#K	 % 
 xx!fr1g"a2q' 2b!Wr1g4FG**
 ''$U4[8q=I 	<I4		
 			qyy  b%077

C''$U4[8q=I<I3		
r8   c                  b   / 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                  " U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                  " U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   .test_reduce_mean_default_axes_keepdims_exampler   r   r!   r"   -test_reduce_mean_default_axes_keepdims_random)r%   r&   r'   r(   r)   r*   r+   r,   r   r.   r/   r0   r1   r2   s         r5   export_default_axes_keepdims'ReduceMean.export_default_axes_keepdims_   s-   xx"((+{{$$F#K	 % 
 xx!fr1g"a2q' 2b!Wr1g4FG**
 ''$THMB 	<IA		
 			qyy  b%077

C''$THMB<I@		
r8   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                  " U[        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                  " U[        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   /test_reduce_mean_negative_axes_keepdims_exampler   r   r!   r"   .test_reduce_mean_negative_axes_keepdims_randomr$   r2   s         r5   export_negative_axes_keepdims(ReduceMean.export_negative_axes_keepdims   s7   xxBHH-{{$$F#K	 % 
 xx!fr1g"a2q' 2b!Wr1g4FG**
 ''$U4[8q=I 	<IB		
 			qyy  b%077

C''$U4[8q=I<IA		
r8    )returnN)
__name__
__module____qualname____firstlineno__staticmethodr6   r<   rA   rG   __static_attributes__rI   r8   r5   r   r      sS    &
 &
P &
 &
P $
 $
L &
 &
r8   r   )numpyr%   r(   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   rI   r8   r5   <module>rT      s!   
   , .a
 a
r8   