
    Si                        S SK Jr  S SKrS SKJr  S SKJr      S           SS jjr " S S\5      r	 " S S	\	5      r
 " S
 S\	5      rg)    )annotationsN)RandomState)OpRunc                6   US:X  d  U(       d/  U(       a%  U [         R                  " U R                  [        S94$ U 4$ [	        U5      nUR                  SSU R                  5      U:  nSSU-
  -  nU(       a  X`-  U-  UR                  [        5      4$ X`-  U-  4$ )Nr   )dtypeg      ?)nponesshapeboolr   uniformastype)Xdrop_probabilityseedtraining_modereturn_maskrndmaskscales           \/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/onnx/reference/ops/op_dropout.py_dropoutr      s     1MbggaggT222t
d
C;;q#qww'+;;D3))*E4?DHudkk$/0XdhQVFVEXX    c                  @    \ rS rSrS r   S         SS jjrSrg)DropoutBase   c                f    [         R                  " XU5        [        UR                  5      U l        g N)r   __init__lenoutput	n_outputs)self	onnx_node
run_paramss      r   r   DropoutBase.__init__   s"    t
3Y--.r   Nc                6    [        UUUU R                  S:H  US9$ )N   )r   r   r   )r   r!   )r"   r   r   ratior   s        r   _private_runDropoutBase._private_run#   s)     !+'
 	
r   )r!   )N      ?F)
r   
np.ndarrayr   
int | Noner(   floatr   r   returntuple[np.ndarray])__name__
__module____qualname____firstlineno__r   r)   __static_attributes__ r   r   r   r      sL    /  #

 
 	

 
 

 
r   r   c                      \ rS rSrSS jrSrg)	Dropout_73   Nc                $    U R                  X5      $ r   )r)   )r"   r   r(   s      r   _runDropout_7._run4   s      **r   r6   r   r1   r2   r3   r4   r;   r5   r6   r   r   r8   r8   3   s    +r   r8   c                  "    \ rS rSrSS.S jrSrg)
Dropout_128   N)r   c                   US   n[        U5      S::  a  SOUS   n[        U5      S::  a  SOUS   nU R                  UUUUS9$ )Nr      r+   r'   F)r   r(   r   )r   r)   )r"   r   inputsr   r(   r   s         r   r;   Dropout_12._run9   s[    1I6{a'VAY!$V!1vay  '	 ! 
 	
r   r6   r=   r6   r   r   r?   r?   8   s    !% 	
 	
r   r?   )r+   NFF)r   r,   r   r.   r   r-   r   r   r   r   r/   r0   )
__future__r   numpyr   numpy.randomr   onnx.reference.op_runr   r   r   r8   r?   r6   r   r   <module>rI      s    #  $ '
 "YYY Y 	Y
 Y Y$
% 
*+ +


 

r   