+
    Pjs                        ^ RI HtHtHt ^ RIt^RIHtHt ^RIH	t	 Rt
 ! R R]]P                  P                  4      t ! R R	]]P                  P                  4      t ! R
 R]	]P                  P                  4      tR# )    )DictOptionalcastN)RearrangeMixinReduceMixin)_EinmixMixinPaddlePaddlec                   &   a  ] tR t^t o R tRtV tR# )	Rearrangec                $    V P                  V4      # N_apply_recipeselfinputs   &&m/Users/ahmed/devFolder/Ultron/claude-voice/gateway/.venv/lib/python3.14/site-packages/einops/layers/paddle.pyforwardRearrange.forward       !!%((     N__name__
__module____qualname____firstlineno__r   __static_attributes____classdictcell____classdict__s   @r   r   r           ) )r   r   c                   &   a  ] tR t^t o R tRtV tR# )Reducec                $    V P                  V4      # r   r   r   s   &&r   r   Reduce.forward   r   r   r   Nr   r    s   @r   r$   r$      r"   r   r$   c                   >   a  ] tR t^t o R tV 3R lR ltR tRtV tR# )EinMixc                $   V P                  V\        P                  P                  P	                  V) V4      R 7      V n        VeD   V P                  V\        P                  P                  P	                  V) V4      R 7      V n        R# RV n        R# ))default_initializerN)create_parameterpaddlenninitializerUniformweightbias)r   weight_shapeweight_bound
bias_shape
bias_bounds   &&&&&r   _create_parametersEinMix._create_parameters   s    ++fii.C.C.K.K\M[g.h , 
 !--		0E0E0M0Mzk[e0f . DI DIr   c          	      r   < V ^8  d   QhRS[ S[,          RS[ S[,          RS[ S[,          RS[ S[,          /# )   pre_reshape_patternpre_reshape_lengthspost_reshape_patternpost_reshape_lengths)r   strr   )formatr!   s   "r   __annotate__EinMix.__annotate__"   sI     f f%c]f &d^f 'sm	f
 'tnfr   c                    R V n         Ve    \        V3/ \        \        V4      B V n         R V n        Ve"   \        V3/ \        \        V4      B V n        R # R # r   )pre_rearranger   r   dictpost_rearrange)r   r:   r;   r<   r=   s   &&&&&r   _create_rearrange_layersEinMix._create_rearrange_layers"   s[     "*!*+>!b$tM`Ba!bD"+"+,@"eDOcDd"eD ,r   c                   V P                   e   V P                  V4      p\        P                  ! V P                  WP                  4      pV P
                  e   W P
                  ,          pV P                  e   V P                  V4      pV# r   )rC   r,   einsumeinsum_patternr0   r1   rE   )r   r   results   && r   r   EinMix.forward1   so    )&&u-Et22E;;G99 iiF*((0Fr   )r1   rE   rC   r0   N)	r   r   r   r   r6   rF   r   r   r   r    s   @r   r(   r(      s      
f f	 	r   r(   )typingr   r   r   r,    r   r   _einmixr   
__author__r-   Layerr   r$   r(   r   r   r   <module>rR      s[    ' '  ) !
)		 )
)[&))// )
%\699?? %r   