+ j,~^RIHt^RIHt]'d ^RIHt^RIHt^RIt!RR4t RR lt R R R llt R#) ) annotations) TYPE_CHECKING)SequenceDimNcz]tRt^t$R]R&RRRlltRRltRR ltR R ltR R lt RRlt RRlt Rt R#)DimEntryz Dim | intdataNc V^8dQhRRRR/#)r zDim | int | NonereturnNone)formats"Y/home/luca/screen-checking/.venv/lib/python3.14/site-packages/functorch/dim/_dim_entry.py __annotate__DimEntry.__annotate__s  -  c ^RIHp\V4\JdV^8d\ RV 24hM/Vf^pM(\ W4'g\ R\V4 24hWnR#)rzExpected negative int, got NzExpected Dim, got )rtypeintAssertionError isinstancer )selfr rs&& r__init__DimEntry.__init__sa : qy$'B4&%IJJ \Dd(($'9$t*%FGG rc V^8dQhRRRR/#)r otherobjectr boolr)rs"rrr!s  F t rc H\V\4'gR#VP4'd0VP4'dVPVP8H#VP4'g/VP4'gVPVPJ#R#F)rr is_positionalr )rr s&&r__eq__DimEntry.__eq__!sy%**     E$7$7$9$999 * *##%%e.A.A.C.C99 * *rcV^8dQhRR/#r r r"r)rs"rrr0s88t8rc f\VP4\J;'dVP^8#r)rr rrs&rr%DimEntry.is_positional0s%DII#%77$))a-7rcV^8dQhRR/#r)r)rs"rrr3s " " "rc h^RIHp\VPV4'dR#VP^8H#)rrF)rrrr )rrs& ris_noneDimEntry.is_none3s) dii % %99> !rcV^8dQhRR/#)r r rr)rs"rrr>s#rc \VP\4'g"\R\ VP4 24hVP#)zExpected int, got )rr rrrr,s&rpositionDimEntry.position>s9$))S)) #5d499o5F!GH HyyrcV^8dQhRR/#)r r rr)rs"rrrCsSrc p\VP\4'd \R4hVP#)zExpected Dim, got int)rr rrr,s&rdim DimEntry.dimCs) dii % % !89 9yyrcV^8dQhRR/#)r r strr)rs"rrrHs#rc ,\VP4#N)reprr r,s&r__repr__DimEntry.__repr__HsDIIr)r r=) __name__ __module__ __qualname____firstlineno____annotations__rr&r%r0r4r8r?__static_attributes__rrrr r s/ O  8 "  rr c V^8dQhRRRR/#)r levelszSequence[DimEntry]r rr)rs"rrrLs  - # rc^^pVF$pVP4'gKV^, pK& V#r+)r%)rHrls& rndim_of_levelsrLLs. A  ??   FA Hrc ,V^8dQhRRRRRRRRRR/#) r tensorz torch.Tensor from_levelszlist[DimEntry] to_levels drop_levelsr"r r)rs"rrrTsA.N.N .N.N.N .N  .NrclW8XdV#VP4pVP4pV'g%\V4\V48d \R4h.p.pVFCpVP V4p VP WI,4VP WY,4KE VPWgVP44# \ dgTP4'dTP ^4M)TP TP4P4TP ^4Ki;i)a" Reshape a tensor to match target levels using as_strided. Args: tensor: Input tensor to reshape from_levels: Current levels of the tensor to_levels: Target levels to match drop_levels: If True, missing dimensions are assumed to have stride 0 Returns: Reshaped tensor z,Cannot expand dimensions without drop_levels) sizestridelenrindexappend ValueErrorr%r8 as_stridedstorage_offset) rNrOrPrQsizesstrides new_sizes new_strideslevelidxs &&&& r _match_levelsraTs$ KKMEmmoG  { c)n , !OP PIK -##E*C   UZ (   w| ,   YV5J5J5L MM """$$  #  !1!12   q !  "sC D3#A D32D3r$) __future__rtypingrcollections.abcrrrtorchr rLrarrrrfs5" ( 99x .N.Nr