o
    wi                     @  sH   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 deZ	dS )    )annotationsN)Base)expectc                   @  sD   e Zd ZedddZedddZedddZedd	d
ZdS )
ReduceMeanreturnNonec                  C  s   g d} t jdgt jd}d}tjjdddgdg|d	}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d}t j|t||dkd}t	|||g|gdd t j
d t j
dd| t j}t j|t||dkd}t	|||g|gd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    r4   c/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/onnx/backend/test/case/node/reducemean.pyexport_do_not_keepdims   :   ,
z!ReduceMean.export_do_not_keepdimsc                  C  s   g d} t jdgt jd}d}tjjdddgdg|d}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d}t j|t||dkd}t	|||g|gdd t j
d t j
dd| t j}t j|t||dkd}t	|||g|gd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#   r1   r4   r4   r5   export_keepdims7   r7   zReduceMean.export_keepdimsc                  C  s   g d} t jg t jd}d}tjjdddgdg|d}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d}t j|d |dkd}t|||g|gdd t j	
d t j	dd| t j}t j|d |dkd}t|||g|gd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.   r/   r0   r1   r4   r4   r5   export_default_axes_keepdims`   s:   ,
z'ReduceMean.export_default_axes_keepdimsc                  C  s   g d} t jdgt jd}d}tjjdddgdg|d	}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d}t j|t||dkd}t	|||g|gdd t j
d t j
dd| t j}t j|t||dkd}t	|||g|gd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#   r1   r4   r4   r5   export_negative_axes_keepdims   r7   z(ReduceMean.export_negative_axes_keepdimsN)r   r   )__name__
__module____qualname__staticmethodr6   r:   r=   rA   r4   r4   r4   r5   r      s    ((&r   )

__future__r   numpyr$   r'   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r4   r4   r4   r5   <module>   s   