U
    qh                     @  s4   d dl mZ d dlZd dlmZ G dd deZdS )    )annotationsN)OpRunc                   @  s   e Zd ZdddZdS )InstanceNormalizationNc                 C  s   t |j}ttd|}tj||dd}tj||dd}d|d  }	|jd|	 }|jd|	 }|||  t||  | }
|
	|j
fS )N   T)axisZkeepdims)   )r   )r   )lenshapetuplerangenpmeanvarZreshapesqrtZastypeZdtype)selfxsZbiasepsilonZdims_xr   r   r   Zdim_onesy r   P/tmp/pip-unpacked-wheel-xnis5xre/onnx/reference/ops/op_instance_normalization.py_run   s    
zInstanceNormalization._run)N)__name__
__module____qualname__r   r   r   r   r   r      s   r   )
__future__r   Znumpyr   Zonnx.reference.op_runr   r   r   r   r   r   <module>   s   