
    /j                    V    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	dS )    )annotationsN)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returnNonec                    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         k/home/longshao/multi-rider-rag/.venv/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&   r4   s         r7   export_keepdimszReduceMean.export_keepdims7   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		
 	
 	
 	
 	
 	
r9   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   r0   r1   r2   r3   r4   s         r7   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@		
 	
 	
 	
 	
 	
r9   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&   r4   s         r7   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		
 	
 	
 	
 	
 	
r9   N)r   r	   )__name__
__module____qualname__staticmethodr8   r=   rA   rF    r9   r7   r   r      s        &
 &
 &
 \&
P &
 &
 &
 \&
P $
 $
 $
 \$
L &
 &
 &
 \&
 &
 &
r9   r   )

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