1;jirdZddlmZmZddlmZmZmZmZddl m Z GddZ Gdd Z d S) zAlex Rogozhnikov)AnyDict)TransformRecipe _apply_recipe_prepare_recipes_for_all_dims get_backend) EinopsErrorcjeZdZdZdededdffd ZdefdZdee e ffdZ d Z d Z d ZxZS) RearrangeMixinz Rearrange layer behaves identically to einops.rearrange operation. :param pattern: str, rearrangement pattern :param axes_lengths: any additional specification of dimensions See einops.rearrange for source_examples. pattern axes_lengthsreturnNc t||_||_||_t |j|_dSN) super__init__r r multirecipe _multirecipetupleitems _axes_lengths)selfr r __class__s P/root/voice-cloning/.venv/lib/python3.11/site-packages/einops/layers/__init__.pyrzRearrangeMixin.__init__sa  ( ,,.."4#4#:#:#<#<==ct|j}|jD]\}}|d||z }d|jj|S)N, {}={}{}({}))reprr rrformatr__name__rparamsaxislengths r__repr__zRearrangeMixin.__repr__shdl## -3355 5 5LD& i&&tV44 4FFt~6???rc t|jdt|jS#t$r(}t d||d}~wwxYw)N rearrange operation axes_names Error while preparing {!r} {})rr rrr r!res rrzRearrangeMixin.multirecipe"sw R0 dFW@X@X  R R R@GGaPPQQ Q Rs(+ A#AAct|}t||jt|j|d|jS)Nr)backendrecipetensorreduction_typer)r rrlenshaperrxr2s rrzRearrangeMixin._apply_recipe*sFa..$S\\2&+     rc |j|jdS)Nr rr;rs r __getstate__zRearrangeMixin.__getstate__4s<9JKKKrc<|jdd|di|ddS)Nr rrrstates r __setstate__zRearrangeMixin.__setstate__7s/ HHeI.H%2GHHHHHrr" __module__ __qualname____doc__strrrr'rintrrrr=rC __classcell__rs@rr r s>>S>T>>>>>>@#@@@@ RT#"67RRRR   LLLIIIIIIIrr cdeZdZdZdededeffd ZdZdee e ffdZ d Z d Z d ZxZS) ReduceMixinaN Reduce layer behaves identically to einops.reduce operation. :param pattern: str, rearrangement pattern :param reduction: one of available reductions ('min', 'max', 'sum', 'mean', 'prod'), case-sensitive :param axes_lengths: any additional specification of dimensions See einops.reduce for source_examples. r reductionrc t||_||_||_||_t|j|_ dSr) rrr rNrrrrrr)rr rNrrs rrzReduceMixin.__init__Fsh  "( ,,.."4#4#:#:#<#<==rcd|j|j}|jD]\}}|d||z }d|jj|S)Nz {!r}, {!r}rr)r!r rNrrrr"r#s rr'zReduceMixin.__repr__Nsr$$T\4>BB -3355 5 5LD& i&&tV44 4FFt~6???rrc t|j|jt|jS#t $r(}t d||d}~wwxYw)Nr*r-)rr rNrrr r!r.s rrzReduceMixin.multirecipeTsy R0 5IZC[C[  R R R@GGaPPQQ Q Rs-0 A"#AA"ct|}t||jt|j||j|jS)Nr1)r rrr6r7rNrr8s rrzReduceMixin._apply_recipe\sHa..$S\\2>+     rc,|j|j|jdS)Nr rNrrTr<s rr=zReduceMixin.__getstate__fs>>S>>>>>>@@@ RT#"67RRRR   iiigggggggrrMN) __author__typingrreinopsrrrr r r rMr?rrrZs  `___________.I.I.I.I.I.I.I.Ib/g/g/g/g/g/g/g/g/g/gr