
    Si                    N    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\5      r	g)    )annotationsN)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   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 )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         e/var/www/html/ai-image-ml/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%   r3   s         r6   export_keepdimsReduceMean.export_keepdims7   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		
r9   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/   r0   r1   r2   r3   s         r6   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@		
r9   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%   r3   s         r6   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		
r9    N)returnNone)
__name__
__module____qualname____firstlineno__staticmethodr7   r=   rB   rH   __static_attributes__rJ   r9   r6   r   r      sS    &
 &
P &
 &
P $
 $
L &
 &
r9   r   )

__future__r   numpyr&   r)   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   rJ   r9   r6   <module>rW      s$    #   , .a
 a
r9   