o
    r::j¨  ã                   @  s>   d dl mZ d dlZd dlmZ ddd„ZG d	d
„ d
eƒZdS )é    )ÚannotationsN)ÚOpRunUnaryNumÚxú
np.ndarrayÚreturnc                 C  s8   | dk}t  | ¡}t  |ddt  |  ¡  |d|  ¡S )zNNumerically stable sigmoid implementation that supports scalars and nd-arrays.r   g      ð?)ÚnpÚexpÚwhere)r   Úpos_maskÚexp_x© r   úd/home/nk/hobo-godmode/plappi-mvp/.venv/lib/python3.10/site-packages/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  | ||¡ d S ©N)r   Ú__init__)ÚselfÚ	onnx_nodeÚ
run_paramsr   r   r   r      ó   zSigmoid.__init__c                 C  s   t |ƒ |j¡fS r   )r   ÚastypeÚdtype)r   ÚXr   r   r   Ú_run   r   zSigmoid._runN)Ú__name__Ú
__module__Ú__qualname__r   r   r   r   r   r   r      s    r   )r   r   r   r   )Ú
__future__r   Únumpyr   Úonnx.reference.ops._opr   r   r   r   r   r   r   Ú<module>   s
   
