
    
Ng                     R    d dl Zd dlmZ  G d de          Z G d de          ZdS )    N)OpRunReduceNumpyc                       e Zd ZddZdS )ReduceMean_1Nc                     |t          |          nd }t          j        ||||j                  }|dk    r.t	          |t          j                  st          j        |          }|fS )Naxiskeepdimsdtyper   )tuplenpmeanr
   
isinstancendarrayarray)selfdataaxesr	   ress        ]/var/www/html/ai-engine/env/lib/python3.11/site-packages/onnx/reference/ops/op_reduce_mean.py_runzReduceMean_1._run   s]    ".uT{{{Dgd
KKKq==C!<!<=(3--Cv    )NN__name__
__module____qualname__r    r   r   r   r      s(             r   r   c                       e Zd ZddZdS )ReduceMean_18N   r   c           
      x   |                      |          r|r|fS |                     |          }|dk    }	 t          j        ||||j                  }|dk    r.t          |t          j                  st          j        |          }|fS # t          $ r$}t          d|j	        d|d| d          |d }~ww xY w)Nr   r   zUnable to reduce shape z with axes=z and keepdims=.)
is_axes_emptyhandle_axesr   r   r
   r   r   r   	TypeErrorshape)r   r   r   r	   noop_with_empty_axesr   es          r   r   zReduceMean_18._run   s    d## 	(< 	7N%%q=		'$THDJOOOC1}}ZRZ%@%@}hsmm6M 	 	 	d$*dd4ddYaddd 	s   AB 
B9B44B9)Nr   r   r   r   r   r   r   r      s(             r   r   )numpyr   onnx.reference.ops._opr   r   r   r   r   r   <module>r*      s        3 3 3 3 3 3    #       $     r   