
    /j                    J    d dl mZ d dlZd dlmZ d
dZ G d d	e          ZdS )    )annotationsN)OpRunUnaryNumx
np.ndarrayreturnc                    | dk    }t          j        |           }t          j        |ddt          j        |            z   z  |d|z   z            S )zNNumerically stable sigmoid implementation that supports scalars and nd-arrays.r   g      ?)npexpwhere)r   pos_maskexp_xs      b/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/onnx/reference/ops/op_sigmoid.pysigmoidr      sR    1uHF1IIE8sRVQBZZ u      c                      e Zd Zd Zd ZdS )Sigmoidc                2    t          j        | ||           d S N)r   __init__)self	onnx_node
run_paramss      r   r   zSigmoid.__init__   s    tY
;;;;;r   c                R    t          |                              |j                  fS r   )r   astypedtype)r   Xs     r   _runzSigmoid._run   s!    

!!!'**,,r   N)__name__
__module____qualname__r   r    r   r   r   r      s2        < < <- - - - -r   r   )r   r   r   r   )
__future__r   numpyr	   onnx.reference.ops._opr   r   r   r!   r   r   <module>r%      s    # " " " " "     0 0 0 0 0 0   - - - - -m - - - - -r   