EiZSSKJr SrSrSrSrSrSrSrS r S r S r S r S r g)) OrderedDictcDUR5(a [S5eg)NzzNYI: Named tensors don't support serialization. Please drop names via `tensor = tensor.rename(None)` before serialization.) has_names RuntimeError)tensors Y/home/mande/repo/quber/.venv/lib/python3.13/site-packages/torch/_namedtensor_internals.pycheck_serializing_named_tensorr s)  M  c[[UR5VVs/sHupUcUOUU4PM snn5$s snnf)zgReturns a map of { dim: dim_name } where dim is a name if the dim is named and the dim index otherwise.)r enumeratenames)ridxnames r build_dim_maprs> @I&,,@WX@W93#4 .@WX Xs; c [U[5(aUR5n[US5(d,[U[5(d[ S[ U535e[U5S:Xa [ S5e[U6$)N__iter__zBExpected namedshape to be OrderedDict or iterable of tuples, got: rz!Expected namedshape to non-empty.) isinstanceritemshasattrtuplertypelenzip) namedshapes runzip_namedshapersz*k**%%' :z * *:j%3P3PPQUV`QaPb c   :!>??  r cU(agg)Nrename_rename)inplaces rnamer_api_namer!(s r c,U[:H=(d US:H$)Nz...)Ellipsis)items r is_ellipsisr%/s 8  ,tu},r c[U5VVs/sHup#[U5(dMUPM nnn[U5S:a[USUS35e[U5S:XaUS$gs snnf)Nz1: More than one Ellipsis ('...') found in names (z-). This function supports up to one Ellipsis.r)r r%rr)r fn_nameirellipsis_indicess rsingle_ellipsis_indexr,3s|)25)9O)9ga[=N)9O !iHgB D   !"" Ps A'A'c"X [U5U- $)N)r)numel_pre_globnumel_post_globr s rexpand_single_ellipsisr0?s #e*"> ??r cZ[U[U5U- S- U5nUSUU-XS-S-$)Nr()r0r) ellipsis_idxr tensor_names globbed_namess rreplace_ellipsis_by_positionr5CsE*c%j</!3\M , - /%q8H8J2K KKr c:[X5nUcU$[X0U5$)zP Expands ... inside `names` to be equal to a list of names from `tensor_names`. )r,r5)r r3r)r2s rresolve_ellipsisr7Js&)8L ' \ JJr c [U5S:XaUScURSU5$UR[XR[ U55U5$)Nr(r)r _update_namesr7r r!)rr r s rupdate_names_with_listr:TsQ 5zQ58+##D'22    nW.EF r c [U5nUH=nXnXC;aXSU'M[[U5SUSUSURSUS3 5e UR [ UR 55U5$)Nz: Tried to rename dim 'z ' to dim z in Tensor[z ] but dim 'z' does not exist)rrr!r r9rvalues)r rename_mapr dim_mapold_dimnew_dims rupdate_names_with_mappingrA^sF#G%  &G !'*++B7)9);v||nKyHXZ     gnn&6 7 AAr c  [U5S:n[U5nU(a6U(a/[[U5S[U5S[U5S35eU(dU(d [ XU5$U(a [ XU5$[ XU5$)aThere are two usages: tensor.rename(*names) returns a view on tensor with named dims `names`. `names` must be of length `tensor.dim()`; otherwise, if '...' is in `names`, then it is expanded greedily to be equal to the corresponding names from `tensor.names`. For example, ``` >>> # xdoctest: +SKIP >>> x = torch.empty(2, 3, 5, 7, names=('N', 'C', 'H', 'W')) >>> x.rename('...', 'height', 'width').names ('N', 'C', 'height', 'width') >>> # xdoctest: +SKIP >>> x.rename('batch', '...', 'width').names ('batch', 'C', 'H', 'width') ``` tensor.rename(**rename_map) returns a view on tensor that has rename dims as specified in the mapping `rename_map`. For example, ``` >>> # xdoctest: +SKIP >>> x = torch.empty(2, 3, 5, 7, names=('N', 'C', 'H', 'W')) >>> x.rename(W='width', H='height').names ('N', 'C', 'height', 'width') ``` Finally, tensor.rename has an in-place version called tensor.rename_. rzW: This function takes either positional args or keyword args, but not both. Use tensor.z!(*names) to name dims and tensor.z(**rename_map) to rename dims.)rboolrr!r:rA)rr r=r rhas_rename_pairss r update_namesrElsFE QIJ'%g&'(>>LW>U=VW''5g'>&?@    -%fW==%fW== $V AAr N) collectionsrr rrr!r%r,r0r5r7r:rArErr rrGsN#  - @LK B3Br