
    bIg                    N    d dl mZ d dlZd dlZd dlmZ d dlmZ  G d de      Z	y)    )annotationsN)Base)expectc                  T    e Zd Zedd       Zedd       Zedd       Zedd       Zy)
ReduceMeanc                    g d} t        j                  dgt         j                        }d}t        j                  j                  dddgdg|	      }t        j                  d
dgddggddgddggddgddgggt         j                        }t        j                  |t        |      |dk(        }t        |||g|gd       t         j                  j                  d       t         j                  j                  dd|       j                  t         j                        }t        j                  |t        |      |dk(        }t        |||g|gd       y )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         c/var/www/html/answerous/venv/lib/python3.12/site-packages/onnx/backend/test/case/node/reducemean.pyexport_do_not_keepdimsz!ReduceMean.export_do_not_keepdims   s;   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                    g d} t        j                  dgt         j                        }d}t        j                  j                  dddgdg|      }t        j                  d	dgd
dggddgddggddgddgggt         j                        }t        j                  |t        |      |dk(        }t        |||g|gd       t         j                  j                  d       t         j                  j                  dd|       j                  t         j                        }t        j                  |t        |      |dk(        }t        |||g|gd       y )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_keepdimszReduceMean.export_keepdims7   s;   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		
r7   c                    g d} t        j                  g t         j                        }d}t        j                  j                  dddgdg|      }t        j                  d	dgd
dggddgddggddgddgggt         j                        }t        j                  |d |dk(        }t        |||g|gd       t         j                  j                  d       t         j                  j                  dd|       j                  t         j                        }t        j                  |d |dk(        }t        |||g|gd       y )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_keepdimsz'ReduceMean.export_default_axes_keepdims`   s1   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@		
r7   c                    g d} t        j                  dgt         j                        }d}t        j                  j                  dddgdg|	      }t        j                  d
dgddggddgddggddgddgggt         j                        }t        j                  |t        |      |dk(        }t        |||g|gd       t         j                  j                  d       t         j                  j                  dd|       j                  t         j                        }t        j                  |t        |      |dk(        }t        |||g|gd       y )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_keepdimsz(ReduceMean.export_negative_axes_keepdims   s;   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		
r7   N)returnNone)__name__
__module____qualname__staticmethodr6   r;   r?   rD    r7   r5   r   r      sS    &
 &
P &
 &
P $
 $
L &
 &
r7   r   )

__future__r   numpyr%   r(   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   rK   r7   r5   <module>rP      s$    #   , .a
 a
r7   