
    IЦis                     ^   S SK r S SKrS SKJrJr  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/                  \R0                  S5      rSr\(       d  SSKJr   " S S5      r\\R8                  4r " S S\	R<                  \5      r " S S\\	R8                  5      rS r\(       aT  \	R@                  r S r!\	RD                  " \	RF                  5      r$\	RJ                  r%\	RD                  " \	RL                  5      r&O=\R@                  \RB                  sr r!\RH                  r$\RJ                  r%\RL                  r&\	RD                  " \	RN                  5      r(\$\l#        \(\l'        \&\R8                  l&        \&\l&        \	RD                  " \	RR                  5      \R8                  l)        \	RD                  " \	RT                  5      \R8                  l*        \" \\\R8                  \RV                  5        \?,\(       a"  \	RD                  " \	RZ                  5      \l-        O\R\                  \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/S0S19  \!" S2S0S19  \!" S3SS'9  \!" S4SS0S59  \!" S6SS0S79  \!" S8SS0S79  \!" S9SS0S79  \!" S:SS0S79  \!" S;SS0S79  \!" S<SS0S79  \!" S=SS0S79  \!" S>SSS0S?9  \!" S@SS0S79  \ " \R^                  R`                  Rb                  SS0S79r1g)A    N)SequenceUnion)dim   )tree_flattentree_map)	wrap_typec                       \ rS rSrSrg)DimensionMismatchError    N__name__
__module____qualname____firstlineno____static_attributes__r       U/var/www/html/ai-image-ml/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_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>:   s*     +b[aVW
1c8J8JAHPQ,Q[as   ),z sizes=)_tensorr)   r/   r(   size)r+   tensorlevelsr/   s      @r   __repr___Tensor.__repr__8   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   indexr:   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)TensorF   r   N)r   r   r   r   r=   staticmethodr?   Tensor_from_batchedfrom_batchedTensor_from_positionalfrom_positionalrA   
Tensor_sumsumr   r   r   r   rI   rI   F   s9    #B$:$:;"2#<#<=O


R]]
+Cr   rI   c                 P    [        5       n[        XU5      R                  X1/U5      $ r!   )r,   stackrC   )tensorsr   new_dimns       r   catrW   M   s%    AS!'''::r   c                     [        [        R                  U 5      n[        [        U [
        R                  " [        U/UQ70 UD65      5        g r!   )getattrtorchrI   setattrr   r?   rA   _wrap)nameargskwargsorigs       r   _defra   U   s8    u||T*r11%2Nt2Nv2NOPr   meanrQ   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_offsetr{   cummax)rr   r{   cummincumprodcumprod_cumsumcumsum_logcumsumexprenorm)r   rr   r{   softmax)2disinspecttypingr   r   functorch._C	functorchrZ   r   r?   r   r   r	   _patch_tensor_classr,   DimListdimlists	Exceptionr   r    r   dictfromkeys	pointwiser=   r   r   rI   
TensorLiker#   rW   r\   ra   rA   __getitem__t__getitem__rS   split__setitem__t__setitem__rB   rC   r@   r/   order
positionalnn
functionalr   r   r   r   <module>r      sg   
  "   " ,      ''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   