U
    q±Ëh¢  ã                   @  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„ ZdS )ÚPReluc                 C  sÊ   z"t  |dk||| ¡ |j¡fW S  tk
rÄ   t|jƒdkr¾|jd }g }d}|jD ]*}||krx| |¡ |d7 }qX| d¡ qX|dkr¾|| t	|ƒ¡ }t  |dk||¡ |j¡f Y S ‚ Y nX d S )Nr   é   )
ÚnpÚwhereZastypeZdtypeÚ
ValueErrorÚlenÚshapeÚappendZreshapeÚtuple)ÚselfÚxZslopeZdimZ	new_shapeÚnÚdÚxs© r   ú?/tmp/pip-unpacked-wheel-xnis5xre/onnx/reference/ops/op_prelu.pyÚ_run   s     "



 z
PRelu._runN)Ú__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   