
    ёi	                        S SK JrJrJr  SSKJr  / r " S S\5      r " S S\5      r " S S	\5      r	 " S
 S\5      r
 " S S\5      r " S S\5      r " S S\5      r " S S\5      r " S S\5      r " S S\5      rg)   )linalgmanipulationmath   )Layerc                   (   ^  \ rS rSrU 4S jrSrU =r$ )FloatFunctionalLayer   c                 "   > [         TU ]  5         g Nsuper__init__self	__class__s    a/var/www/html/banglarbhumi/venv/lib/python3.13/site-packages/paddle/nn/quant/functional_layers.pyr   FloatFunctionalLayer.__init__            )__name__
__module____qualname____firstlineno__r   __static_attributes____classcell__r   s   @r   r	   r	      s     r   r	   c                   2   ^  \ rS rSrU 4S jrSS jrSrU =r$ )add   c                 "   > [         TU ]  5         g r   r   r   s    r   r   add.__init__   r   r   c                 ,    [         R                  " XUS9$ N)name)r   r    r   xyr&   s       r   forwardadd.forward   s    xx4((r   r   r   r   r   r   r   r   r*   r   r   r   s   @r   r    r       s    ) )r   r    c                   2   ^  \ rS rSrU 4S jrSS jrSrU =r$ )subtract"   c                 "   > [         TU ]  5         g r   r   r   s    r   r   subtract.__init__#   r   r   c                 ,    [         R                  " XUS9$ r%   )r   r.   r'   s       r   r*   subtract.forward&       }}Q--r   r   r   r,   r   s   @r   r.   r.   "       . .r   r.   c                   2   ^  \ rS rSrU 4S jrSS jrSrU =r$ )multiply*   c                 "   > [         TU ]  5         g r   r   r   s    r   r   multiply.__init__+   r   r   c                 ,    [         R                  " XUS9$ r%   )r   r7   r'   s       r   r*   multiply.forward.   r4   r   r   r   r,   r   s   @r   r7   r7   *   r5   r   r7   c                   2   ^  \ rS rSrU 4S jrSS jrSrU =r$ )divide2   c                 "   > [         TU ]  5         g r   r   r   s    r   r   divide.__init__3   r   r   c                 ,    [         R                  " XUS9$ r%   )r   r>   r'   s       r   r*   divide.forward6   s    {{1d++r   r   r   r,   r   s   @r   r>   r>   2   s    , ,r   r>   c                   2   ^  \ rS rSrU 4S jrSS jrSrU =r$ )reshape:   c                 "   > [         TU ]  5         g r   r   r   s    r   r   reshape.__init__;   r   r   c                 ,    [         R                  " XUS9$ r%   )r   rE   )r   r(   shaper&   s       r   r*   reshape.forward>   s    ##A488r   r   r   r,   r   s   @r   rE   rE   :   s    9 9r   rE   c                   2   ^  \ rS rSrU 4S jrSS jrSrU =r$ )	transposeB   c                 "   > [         TU ]  5         g r   r   r   s    r   r   transpose.__init__C   r   r   c                 ,    [         R                  " XUS9$ r%   )r   rM   )r   r(   permr&   s       r   r*   transpose.forwardF   s    %%aD99r   r   r   r,   r   s   @r   rM   rM   B   s    : :r   rM   c                   2   ^  \ rS rSrU 4S jrSS jrSrU =r$ )concatJ   c                 "   > [         TU ]  5         g r   r   r   s    r   r   concat.__init__K   r   r   c                 ,    [         R                  " XUS9$ r%   )r   rU   )r   r(   axisr&   s       r   r*   concat.forwardN   s    ""166r   r   )    Nr,   r   s   @r   rU   rU   J   s    7 7r   rU   c                   2   ^  \ rS rSrU 4S jrSS jrSrU =r$ )flattenR   c                 "   > [         TU ]  5         g r   r   r   s    r   r   flatten.__init__S   r   r   c                 ,    [         R                  " XX4S9$ r%   )r   r^   )r   r(   
start_axis	stop_axisr&   s        r   r*   flatten.forwardV   s    ##A9HHr   r   )r\   Nr,   r   s   @r   r^   r^   R   s    I Ir   r^   c                   2   ^  \ rS rSrU 4S jrSS jrSrU =r$ )matmulZ   c                 "   > [         TU ]  5         g r   r   r   s    r   r   matmul.__init__[   r   r   c                 .    [         R                  " XX4US9$ r%   )r   rh   )r   r(   r)   transpose_xtranspose_yr&   s         r   r*   matmul.forward^   s    }}Q;$GGr   r   )FFNr,   r   s   @r   rh   rh   Z   s    H Hr   rh   N)tensorr   r   r   layer.layersr   __all__r	   r    r.   r7   r>   rE   rM   rU   r^   rh   r   r   r   <module>rs      s    1 0  
5 
)
 ).# ..# .,! ,9" 9:$ :7! 7I" IH! Hr   