
    bIg                    P    d dl mZ d dlZd dlmZ  G d de      Z G d de      Zy)    )annotationsN)OpRunReduceNumpyc                      e Zd ZddZy)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/answerous/venv/lib/python3.12/site-packages/onnx/reference/ops/op_reduce_mean.py_runzReduceMean_1._run   sR    ".uT{Dggd

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y)ReduceMean_18Nc           
     r   | j                  |      r|r|fS | j                  |      }|dk7  }	 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=	''$THDJJOC1}ZRZZ%@hhsm
 v	  	)$**{4(.YaXbbcd	s   AB 	B6 B11B6)N   r   r   r   r   r   r   r      s    r   r   )
__future__r   numpyr   onnx.reference.ops._opr   r   r   r   r   r   <module>r,      s*    #  3# $ r   