U
    q±ËhÌ  ã                   @  s<   d dl mZ d dlZd dlmZ dd„ ZG dd„ deƒZdS )é    )ÚannotationsN)ÚOpRunUnaryNumc                 C  s4   | dkrddt  |  ¡  S t  | ¡dt  | ¡  S )Nr   é   )ÚnpÚexp)Úx© r   úA/tmp/pip-unpacked-wheel-xnis5xre/onnx/reference/ops/op_sigmoid.pyÚsigmoid   s    r
   c                   @  s   e Zd Zdd„ Zdd„ ZdS )ÚSigmoidc                 C  s   t  | ||¡ t t¡| _d S )N)r   Ú__init__r   Z	vectorizer
   Úvf)ÚselfZ	onnx_nodeZ
run_paramsr   r   r	   r      s    zSigmoid.__init__c                 C  sD   t |jƒdkr t|ƒ |j¡fS |jdkr0|fS |  |¡ |j¡fS )Nr   )ÚlenÚshaper
   ZastypeZdtypeÚsizer   )r   ÚXr   r   r	   Ú_run   s
    
zSigmoid._runN)Ú__name__Ú
__module__Ú__qualname__r   r   r   r   r   r	   r      s   r   )Ú
__future__r   Znumpyr   Zonnx.reference.ops._opr   r
   r   r   r   r   r	   Ú<module>   s   