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)OpRunUnaryNumc                   @  s   e Zd ZdddZdS )LpNormalizationNc                 C  sR   |p| j }|p| j}tt||j|dd| }t||}|| |jfS )N)axisg      ?)r   pnppowersumZexpand_dimsZastypeZdtype)selfxr   r   Znorm r   J/tmp/pip-unpacked-wheel-xnis5xre/onnx/reference/ops/op_lp_normalization.py_run   s
    

 zLpNormalization._run)NN)__name__
__module____qualname__r   r   r   r   r   r      s   r   )
__future__r   Znumpyr   Zonnx.reference.ops._opr   r   r   r   r   r   <module>   s   