
    Ng                     J    d dl Zd dlZd dlmZ d dlmZ  G d de          ZdS )    N)Base)expectc                   n    e Zd Zedd            Zedd            Zedd            Zedd            ZdS )
ReduceMeanreturnNc                     g d} t          j        dgt           j                  }d}t          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
                            d           t           j
                            dd|                               t           j                  }t          j        |t          |          |dk              }t          |||g|gd           d S )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         b/var/www/html/ai-engine/env/lib/python3.11/site-packages/onnx/backend/test/case/node/reducemean.pyexport_do_not_keepdimsz!ReduceMean.export_do_not_keepdims   s   		x28,,,{$$F#K	 % 
 
 x!fr1g"a2q' 2b!Wr1g4FG*
 
 
 '$U4[[8q=III 	$<I;		
 	
 	
 	
 		qy  b%0077
CC'$U4[[8q=III$<I:		
 	
 	
 	
 	
 	
    c                     g d} t          j        dgt           j                  }d}t          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
                            d           t           j
                            dd|                               t           j                  }t          j        |t          |          |dk              }t          |||g|gd           d S )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_keepdims6   s   		x28,,,{$$F#K	 % 
 
 x!fr1g"a2q' 2b!Wr1g4FG*
 
 
 '$U4[[8q=III 	$<I4		
 	
 	
 	
 		qy  b%0077
CC'$U4[[8q=III$<I3		
 	
 	
 	
 	
 	
r7   c                     g d} t          j        g t           j                  }d}t          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	        
                    d           t           j	                            dd|                               t           j                  }t          j        |d |dk              }t          |||g|gd           d S )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_   sv   		x"(+++{$$F#K	 % 
 
 x!fr1g"a2q' 2b!Wr1g4FG*
 
 
 '$THMBBB 	$<IA		
 	
 	
 	
 		qy  b%0077
CC'$THMBBB$<I@		
 	
 	
 	
 	
 	
r7   c                     g d} t          j        dgt           j                  }d}t          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
                            d           t           j
                            dd|                               t           j                  }t          j        |t          |          |dk              }t          |||g|gd           d S )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   		xBH---{$$F#K	 % 
 
 x!fr1g"a2q' 2b!Wr1g4FG*
 
 
 '$U4[[8q=III 	$<IB		
 	
 	
 	
 		qy  b%0077
CC'$U4[[8q=III$<IA		
 	
 	
 	
 	
 	
r7   )r   N)__name__
__module____qualname__staticmethodr6   r;   r?   rD    r7   r5   r   r      s        &
 &
 &
 \&
P &
 &
 &
 \&
P $
 $
 $
 \$
L &
 &
 &
 \&
 &
 &
r7   r   )numpyr%   r(   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   rI   r7   r5   <module>rM      s~   
      , , , , , , . . . . . .a
 a
 a
 a
 a
 a
 a
 a
 a
 a
r7   