Nj>SSKJr SSKJrJr \(a SSKJr SSKJr SSK r "SS5r S S jr S S S jjr g)) annotations) TYPE_CHECKINGUnion)SequenceDimNcj\rSrSr%S\S'S SSjjrSSjrSSjrSSjrSS jr SS jr SS jr S r g)DimEntryzUnion[Dim, int]dataNcSSKJn [U5[LaUS:a[ SU35eO-UcSnO'[ X5(d[ S[U535eXlg)NrrrzExpected negative int, got zExpected Dim, got )r typeintAssertionError isinstancer )selfr r s i/root/GenerationalWealth/GenerationalWealth/venv/lib/python3.13/site-packages/functorch/dim/_dim_entry.py__init__DimEntry.__init__sa : qy$'B4&%IJJ \Dd(($'9$t*%FGG c:[U[5(dgUR5(a.UR5(aURUR:H$UR5(d-UR5(dURURL$g)NF)rr is_positionalr )rothers r__eq__DimEntry.__eq__!sy%**     E$7$7$9$999 * *##%%e.A.A.C.C99 * *rcd[UR5[L=(a URS:$)Nr)rr rrs rrDimEntry.is_positional0s"DII#%7$))a-7rcdSSKJn [URU5(agURS:H$)NrrFr)rr rr )rr s ris_noneDimEntry.is_none3s) dii % %99> !rc[UR[5(d![S[ UR535eUR$)NzExpected int, got )rr rrrrs rpositionDimEntry.position>s9$))S)) #5d499o5F!GH Hyyrcn[UR[5(a [S5eUR$)NzExpected Dim, got int)rr rrrs rdim DimEntry.dimCs) dii % % !89 9yyrc,[UR5$N)reprr rs r__repr__DimEntry.__repr__HsDIIr)r r+)r zUnion[Dim, int, None]returnNone)robjectr/bool)r/r2)r/r)r/r )r/str) __name__ __module__ __qualname____firstlineno____annotations__rrrr"r%r(r-__static_attributes__rrr r s+   8 "  rr cTSnUHnUR5(dMUS- nM! U$)Nrr)r)levelsrls rndim_of_levelsr?Ls. A  ??   FA HrcRX:XaU$UR5nUR5nU(d#[U5[U5:a [S5e/n/nUH;nUR U5n UR XI5 UR XY5 M= URXgUR55$![ ae UR5(aUR S5 O)UR UR5R5 UR S5 Mf=f)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_levelsrr) sizestridelenrindexappend ValueErrorrr( as_stridedstorage_offset) tensor from_levels to_levels drop_levelssizesstrides new_sizes new_strideslevelidxs r _match_levelsrSTs$ KKMEmmoG  { c)n , !OP PIK -##E*C   UZ (   w| ,   YV5J5J5L MM """$$  #  !1!12   q !  "sB77A+D&%D&)r<zSequence[DimEntry]r/r)F) rI torch.TensorrJlist[DimEntry]rKrUrLr2r/rT) __future__rtypingrrcollections.abcrrr torchr r?rSr:rrrZsg"'( 99x  .N .N.N.N .N  .Nr