
    9i0                     b   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 S5      r\\R*                  4r " S S\R.                  \5      r " S S\\R*                  5      rS r\R2                  rS r\R6                  " \R8                  5      r\R<                  r\R6                  " \R>                  5      r\R6                  " \R@                  5      r!\\l        \!\l         \\R*                  l        \\l        \R6                  " \RD                  5      \R*                  l"        \R6                  " \RF                  5      \R*                  l#        \" \\R*                  \RH                  5        \?%\R6                  " \RL                  5      \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.S/9  \" S0S.S/9  \" S1SS%9  \" S2SS.S39  \" S4SS.S59  \" S6SS.S59  \" S7SS.S59  \" S8SS.S59  \" S9SS.S59  \" S:SS.S59  \" S;SS.S59  \" S<SSS.S=9  \" S>SS.S59  \" \RN                  RP                  RR                  SS.S59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       V/var/www/html/land-doc-ocr/venv/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_propertiesTc                       \ rS rSr\S 5       rS r\" \R                  5      r	\R                  " \R                  5      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>.   s*     +b[aVW
1c8J8JAHPQ,Q[as   ),z sizes=)_tensorr&   r,   r%   size)r(   tensorlevelsr,   s      @r   __repr___Tensor.__repr__,   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   classmethod_C__torch_function___instancemethodexpandindexr7   r   r   r   r   r   r      s^     D D %R%:%:;		*Frxx(EBr   r   c                   ,    \ rS rSr\R
                  rSrg)r    4   r   N)r   r   r   r   object
__format__r   r   r   r   r    r    4   s     ""Jr   r    c                   p    \ rS rSr\" \R                  5      r\R                  " \R                  5      r
Srg)Tensor:   r   N)r   r   r   r   staticmethodr;   Tensor_from_positionalfrom_positionalr=   
Tensor_sumsumr   r   r   r   rE   rE   :   s&    "2#<#<=O


R]]
+Cr   rE   c                 P    [        5       n[        XU5      R                  X1/U5      $ r   )r)   stackr?   )tensorsr   new_dimns       r   catrQ   ?   s%    AS!'''::r   c                     [        [        R                  U 5      n[        [        U [
        R                  " [        U/UQ70 UD65      5        g r   )getattrtorchrE   setattrr   r;   r=   _wrap)nameargskwargsorigs       r   _defr[   G   s8    5<<&DGT2--eD.J4.J6.JKLr   meanrK   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_offsetru   cummax)rl   ru   cummincumprodcumprod_cumsumcumsum_logcumsumexprenorm)ry   rl   ru   softmax)*functorch._C	functorchrT   r   r;   r   r   r   _patch_tensor_classr)   DimListdimlists	Exceptionr	   r    r   dictfromkeys	pointwiser   rE   
TensorLiker    rQ   rV   r[   r=   __getitem__t__getitem__rM   split__setitem__t__setitem__r>   r?   r<   r,   ordernn
functionalr   r   r   r   <module>r      s     " ,      ''2::r{{ gx	Y 		 	  MM-1148	B B* u||$
#"&&' #,Wbii ,
;
 	M
 !!"..1

288$ !!"..1
 # "  ((3 ''1  	'5<<!;!; <L""288, 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   