U
    rhW                     @  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                   @  s   e Zd ZeddddZdS )MeanNone)returnc                  C  s  t dddgt j} t dddgt j}t dddgt j}t dddgt j}tjjddd	d
gdgd}t|| ||g|gdd tjjddgdgd}t|| g| gdd t t 	| |d}tjjddd	gdgd}t|| |g|gdd d S )N   r               r   data_0data_1data_2result)inputsoutputsZtest_mean_example)r   r   nameZtest_mean_one_inputg       @Ztest_mean_two_inputs)
nparrayZastypeZfloat32onnxhelperZ	make_noder   divideadd)r   r   r   r   node r   D/tmp/pip-unpacked-wheel-xnis5xre/onnx/backend/test/case/node/mean.pyexport   sB       zMean.exportN)__name__
__module____qualname__staticmethodr   r   r   r   r   r      s   r   )

__future__r   Znumpyr   r   Zonnx.backend.test.case.baser   Zonnx.backend.test.case.noder   r   r   r   r   r   <module>   s
   