o
    soi                     @  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ZdS )MeanVarianceNormalizationreturnNonec                  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;?)dtype)r            )axiskeepdimsr   g&.>test_mvn)r
   r   name)
onnxhelper	make_nodenparrayfloat32meanpowersqrtr   )node
input_data	data_meandata_mean_squareddata_squareddata_squared_meanstdexpected_output r%   i/home/ubuntu/.local/lib/python3.10/site-packages/onnx/backend/test/case/node/meanvariancenormalization.pyexport   s4   
z MeanVarianceNormalization.exportN)r   r   )__name__
__module____qualname__staticmethodr'   r%   r%   r%   r&   r      s    r   )

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