+ jcv^RIHt^RIHtHt]'d^RIHt^RIt^RIH t H t H t R RRllt RR lt R#) ) annotations)Any TYPE_CHECKING)SequenceN) _match_levelsDimEntryndim_of_levelsc(V^8dQhRRRRRRRR/#) argr orig_ndimint allow_noneboolreturnr)formats"U/home/luca/screen-checking/.venv/lib/python3.14/site-packages/functorch/dim/_order.py __annotate__rs(33DHc^RIHpVfV'd \4#\W4'd \V4#\V\4'dV^8dTpMW, p\V4#\4#)a Convert various dimension representations to DimEntry. Args: arg: The argument to convert (Dim, int, or other) orig_ndim: Original number of dimensions allow_none: Whether to allow None values Returns: DimEntry representation of the dimension )Dim)rr isinstancer)r r rrposs&&& r _wrap_dimrs] {zz C  } C   7C/C}zrc$V^8dQhRRRRRR/#)r tensor_or_dimztorch.Tensor | AnydimszAny | Sequence[Any]rz torch.Tensorr)rs"rrr*s)mm%m.Ammrc aaa ^RIHpHpHp\ W4'd-VP R,pVP pVPpM;\ W4'd \V4.pVP4pRpM \R4h.o.pVR,o\S4o RVVV 3Rllp ^p VEFp \V S R4p V P4'gV !V 4V ^, p K:\ W4'd0V PFp V !\V 44V ^, p K KzV ^, p \V R4'g \R4h\!V 4pVP#\%S4\%V434VF9p\VS R4p V P4'd \R 4hV !V 4K; EK R p.pSFepVP4'dKVP'4'd$VR 8Xd\%V4pVP)S4VP#V4Kg VR 8Xd\%V4pVP)S4Vf \+R 4h\-WeV4pV'Ed.pVP/4p\1V4FpVP#VV,4K ^pVFwppVV8d+VP#VVV,,4V^, pK1^p\1V4F!pVVVV,V,,,pK# VP#V4VV, pK V\%S48d+VP#VVV,,4V^, pK:\1V\%S4,\%S44FpVP#VV,4K \%S4V , pV^8dVR VVVV,R ,pVP3V4p^p\1\%V4^, R R 4FPpVV,P'4'gVV8gK)VVV ,8gK9V^,p\V4VV&KR VP5VVV4pV#) a Reorder the dimensions of a tensor or create a tensor from a dimension. It allows reordering tensor dimensions using first-class dimensions and positional indices. Args: tensor_or_dim: Input tensor with first-class dimensions, or a Dim object *dims: Dimensions or sequences of dimensions specifying the new order Returns: Tensor with reordered dimensions Examples: >>> import torch >>> from functorch.dim import dims >>> batch, channel, height, width = dims(4) >>> x = torch.randn(2, 3, 4, 5)[batch, channel, height, width] >>> # Reorder to [height, width, batch, channel] >>> y = order(x, height, width, batch, channel) )rDimListTensor:NNNFz-First argument must be a Tensor or Dim objectc V^8dQhRRRR/#)r drrNoner)rs"rrorder..__annotate__Xs''h'4'rcJ<SPV4pVf[VP4'd(\RS RVP4S, R24h\RVP 4 R24h\ 4SV&SP V4R# \dRpLi;i)zEAdd a dimension to the reordering, removing it from available levels.Nz tensor has z positional dimensions, but z% specified, or it was specified twiceztensor does not contain dim z or it was specified twice)index ValueError is_positionalpositiondimrappend)r$idxflat_positional_dimslevelsr s& r append_dimorder..append_dimXs ,,q/C ;   !),HXaIaHbc01 !21557);UVjs ##A& C sB B"!B"__iter__z+expected a Dim, List[Dim], or Sequence[Dim]zexpected a Dim or intNzCannot reorder None tensor)rrr!r"r_levels_tensor _has_devicer _get_ranger)r ris_none_dimshasattrlistr-lenr*extendAssertionErrorrsizerangereshapefrom_positional)!rrrr!r" orig_levelsdata has_device to_flattenr1n_new_positionalr entryr,seqitem insert_point new_levelslevelndata view_shapesizesi start_idxlengthnew_sizej n_to_removeseenresultr/r0r s!&* @@@rorderrZ*s"0'&-((#++A. $$".. M ' ' ./ '') HIIJ ^Fv&I''(#y%0}} u   !   % %yy8C=) A% !  ! 3 ++ !NOOs)C   s#78#c(C D!$ 59==??$%<==5! )6L!#J ==??     r!": !!"67% r: ./ |9:: $Z 8Ez  |$A   eAh '% !+ Ivi-!!% q(8"9:QH6]E,"2Q"677#   h ' KA",#*++   eL1$45 6 FA|c*>&??VMA   eAh 'N./2BB ?=L)J|k7Q7S,TT  j) D 3z?Q&B / a= & & ( (  !l5E&E"E AID$TNJqM 0 # #E:z BF Mr)T) __future__rtypingrrcollections.abcrtorch _dim_entryrrr rrZrrrr`s'"%( ??8mr