
    Ch                    B    S SK Jr  S SKJrJr  S SKJr   " S S\5      rg)    )annotations)Tensornn)Modulec                  b   ^  \ rS rSr% SrS/rS\S'   SSU 4S jjjrSS jrSS	.SS
 jjr	Sr
U =r$ )Dropout   zIDropout layer.

Args:
    dropout: Sets a dropout value for dense layer.
dropoutz	list[str]config_keysc                x   > [         TU ]  5         Xl        [        R                  " U R                  5      U l        g N)super__init__r
   r   r   dropout_layer)selfr
   	__class__s     \/var/www/html/shao/venv/lib/python3.13/site-packages/sentence_transformers/models/Dropout.pyr   Dropout.__init__   s(    ZZ5    c                P    UR                  SU R                  US   5      05        U$ )Nsentence_embedding)updater   )r   featuress     r   forwardDropout.forward   s*    -t/A/A(K_B`/abcr   T)safe_serializationc               &    U R                  U5        g r   )save_config)r   output_pathr   argskwargss        r   saveDropout.save   s    %r   )r
   r   )g?)r
   float)r   zdict[str, Tensor])r   strr   boolreturnNone)__name__
__module____qualname____firstlineno____doc__r   __annotations__r   r   r"   __static_attributes____classcell__)r   s   @r   r   r      s6     ([K(6 6
 HL & &r   r   N)
__future__r   torchr   r   #sentence_transformers.models.Moduler   r    r   r   <module>r5      s    "  6&f &r   