
    Ch.                        S SK Jr   S SKJr  S SKJs  Jr	  S SK
Jr  S SKJr   " S S\5      rg! \ a	    S SKJr   N/f = f)    )annotations)SelfN)Tensor)Modulec                  `   ^  \ rS rSrSrS
U 4S jjrSS jrSS.SS jjr\SS j5       r	S	r
U =r$ )	Normalize   z/This layer normalizes embeddings to unit lengthc                "   > [         TU ]  5         g N)super__init__)self	__class__s    ^/var/www/html/shao/venv/lib/python3.13/site-packages/sentence_transformers/models/Normalize.pyr   Normalize.__init__   s        c           	     Z    UR                  S[        R                  " US   SSS905        U$ )Nsentence_embedding      )pdim)updateF	normalize)r   featuress     r   forwardNormalize.forward   s.    -q{{8DX;Y]^de/fghr   T)safe_serializationc                   g r    )r   output_pathr   argskwargss        r   saveNormalize.save   s    r   c                    U " 5       $ r   r!   )clsr#   r$   s      r   loadNormalize.load   s	    ur   r!   )returnNone)r   dict[str, Tensor]r+   r-   )r"   strr   boolr+   r,   )r+   r   )__name__
__module____qualname____firstlineno____doc__r   r   r%   classmethodr)   __static_attributes____classcell__)r   s   @r   r   r      s0    9 HL   r   r   )
__future__r   typingr   ImportErrortyping_extensionstorch.nn.functionalnn
functionalr   torchr   #sentence_transformers.models.Moduler   r   r!   r   r   <module>rA      s>    "'     6   '&'s   / >>