
    [Th5                     >   S SK rS SKrS SK Jr  SSKJrJr  SSKJr  \R                  " 5         \R                  \R                  \R                  sr	r
r " S S\5      r " S S	\5      rSS
KJr  \R%                  \R&                  S5      rSr\(       d  SSKJr   " S S5      r\\R.                  4r " S S\R2                  \5      r " S S\\R.                  5      rS r\(       aT  \R6                  rS r\R:                  " \R<                  5      r\R@                  r \R:                  " \RB                  5      r!O=\R6                  \R8                  srr\R>                  r\R@                  r \RB                  r!\R:                  " \RD                  5      r#\\l        \#\l"        \!\R.                  l!        \!\l!        \R:                  " \RH                  5      \R.                  l$        \R:                  " \RJ                  5      \R.                  l%        \" \\\R.                  \RL                  5        \?'\(       a"  \R:                  " \RP                  5      \l(        O\RR                  \l(        \" S5        \" S5        \" S5        \" S5        \" S5        \" S5        \" S5        \" S5        \" S5        \" S5        \" S5        \" S 5        \" S!S"S#9  \" S$S"S#9  \" S%SS&9  \" S'SS&9  \" S(SS&9  \" S)SS&9  \" S*SS&9  \" S+SS&9  \" S,SS&9  \" S-SS&9  \" S.S/S09  \" S1S/S09  \" S2SS&9  \" S3SS/S49  \" S5SS/S69  \" S7SS/S69  \" S8SS/S69  \" S9SS/S69  \" S:SS/S69  \" S;SS/S69  \" S<SS/S69  \" S=SSS/S>9  \" S?SS/S69  \" \RT                  RV                  RX                  SS/S69r,g)@    N)dim   )tree_flattentree_map)	wrap_typec                       \ rS rSrSrg)DimensionMismatchError    N__name__
__module____qualname____firstlineno____static_attributes__r       N/var/www/auris/envauris/lib/python3.13/site-packages/functorch/dim/__init__.pyr	   r	          r   r	   c                       \ rS rSrSrg)DimensionBindError   r   Nr   r   r   r   r   r      r   r   r   )op_propertiesT)	referencec                      \ rS rSr\S 5       rS r\(       a/  \" \	R                  5      r
\	R                  " \	R                  5      rO\R                  r
\R                  r\	R                  " \	R                  5      rS rSrg)_Tensor    c                 :    [        S U R                   5       5      $ )Nc              3   T   #    U  H  n[        U[        5      (       d  M  Uv   M      g 7fN)
isinstanceDim).0ds     r   	<genexpr>_Tensor.dims.<locals>.<genexpr>&   s     C1
1c0BQQs   (	()tuple_levelsselfs    r   dims_Tensor.dims$   s    CCCCr   c                     U R                   $ r   )ndimr(   s    r   r   _Tensor.dim(   s    yyr   c                    ^ U R                   U R                  U R                  spmU S[        U4S jU 5       5       S[        UR	                  5       5       3$ )Nz
with dims=c              3   \   >#    U  H!  n[        U[        5      (       a  UT-   OUv   M#     g 7fr   )r    int)r"   lr-   s     r   r$   #_Tensor.__repr__.<locals>.<genexpr>6   s*     +b[aVW
1c8J8JAHPQ,Q[as   ),z sizes=)_tensorr'   r-   r&   size)r)   tensorlevelsr-   s      @r   __repr___Tensor.__repr__4   s]    #||T\\499e+b[a+b&b%ccjkpqwq|q|q~k  kA  B  	Br   r   N)r   r   r   r   propertyr*   r   use_cclassmethod_C__torch_function___instancemethodexpandr   indexr8   r   r   r   r   r   r       sx     D D ()>)>?##BII.&99!!rxx(EBr   r   c                   ,    \ rS rSr\R
                  rSrg)r!   <   r   N)r   r   r   r   object
__format__r   r   r   r   r!   r!   <   s     ""Jr   r!   c                       \ rS rSr\(       d  \" \R                  5      r\" \R                  5      r
\R                  " \R                  5      rSrg)TensorB   r   N)r   r   r   r   r;   staticmethodr=   Tensor_from_batchedfrom_batchedTensor_from_positionalfrom_positionalr?   
Tensor_sumsumr   r   r   r   rG   rG   B   s9    #B$:$:;"2#<#<=O


R]]
+Cr   rG   c                 P    [        5       n[        XU5      R                  X1/U5      $ r   )r*   stackrA   )tensorsr   new_dimns       r   catrU   I   s%    AS!'''::r   c                     [        [        R                  U 5      n[        [        U [
        R                  " [        U/UQ70 UD65      5        g r   )getattrtorchrG   setattrr   r=   r?   _wrap)nameargskwargsorigs       r   _defr_   Q   s8    u||T*r11%2Nt2Nv2NOPr   meanrO   allamaxaminaminmaxanycount_nonzero	logsumexpnanmeannansumprodstd   )keepdim_offsetvarmax)
single_dimminargmaxargminkthvaluemedian	nanmedianmodesortF)reduceargsortunbindchunk)
dim_offsetry   cummax)rp   ry   cummincumprodcumprod_cumsumcumsum_logcumsumexprenorm)r}   rp   ry   softmax)-functorch._C	functorchrX   r   r=   r   r   r   _patch_tensor_classr*   DimListdimlists	Exceptionr	   r    r   dictfromkeys	pointwiser;   r   r   rG   
TensorLiker!   rU   rZ   r_   r?   __getitem__t__getitem__rQ   split__setitem__t__setitem__r@   rA   r>   r-   order
positionalnn
functionalr   r   r   r   <module>r      s^     " ,      ''2::r{{ gx	Y 		 	  MM-1148	B B2 u||$
#"&&' #,Wbii ,;
 	HHEQ %%bnn5LHHErxx(E//9>>KE4))LOOEOOE !!"..1
 # "  ((3 ''1  	%%,,(B(B CL&&rxx0GM((GM V U U V V Y U _  [  Y X V U1  U1  Ut  Ut  X$  X$  ZD ! X$  [T " V  VE  Yu  X$  W5 ) X$u - X$u - Y4 . ZD / X$u - Y4 . ^U 3 X!U ; Y4 .
##++U
Kr   