U
    rh                     @  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 )MeanVarianceNormalizationNone)returnc                  C  s  t jjddgdgd} tjdgdgdggdgd	gd
ggdgdgdgggdgdgdggdgdgdggdgdgdgggdgdgdggdgdgdggdgdgdggggtjd }tj|d!d"d#}t|d$}t|d$}tj|d!d"d#}t|| }|| |d%  }t	| |g|gd&d' d S )(Nr   XY)inputsoutputsg?ghі ?gjFS?g}ݝ?gAH!?g$79P1?gY"s?gJO?ggLw?g8?ge?gX[}?gQ?gFj-_?g>b?g͗r?g?g3*ס?g3w?gH6?gЊQ?g{?7?gy)?g825?gu> (?gpǂ_vQ?g;?)Zdtype)r            )ZaxisZkeepdimsr   g&.>Ztest_mvn)r
   r   name)
onnxhelperZ	make_nodenparrayZfloat32Zmeanpowersqrtr   )nodeZ
input_dataZ	data_meanZdata_mean_squaredZdata_squaredZdata_squared_meanZstdZexpected_output r   Y/tmp/pip-unpacked-wheel-xnis5xre/onnx/backend/test/case/node/meanvariancenormalization.pyexport   s8      z MeanVarianceNormalization.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
   