jI\ddlmZ dZdZdZdZdZdZdZd Z d Z d Z d Z d Z dS)) OrderedDictcL|rtddS)NzzNYI: Named tensors don't support serialization. Please drop names via `tensor = tensor.rename(None)` before serialization.) has_names RuntimeErrortensors MC:\Projects\Historical\venv\Lib\site-packages\torch/_namedtensor_internals.pycheck_serializing_named_tensorr s8   M     cXtdt|jDS)zkReturns a map of { dim: dim_name } where dim is a name if the dim is named and the dim index otherwise.c$g|] \}}||n||fSN).0idxnames r z!build_dim_map..s)XXX93##4 .XXXr )r enumeratenamesrs r build_dim_maprs1 XX &,@W@WXXX  r c2t|tr|}t|ds4t|tst dt |t|dkrt dt|S)N__iter__zBExpected namedshape to be OrderedDict or iterable of tuples, got: rz!Expected namedshape to non-empty.) isinstanceritemshasattrtuplertypelenzip) namedshapes r unzip_namedshaper!s*k**(%%'' :z * * :j%3P3P  cQUV`QaQa c c    :!>???  r c|rdSdS)Nrename_renamer)inplaces r namer_api_namer&(syxr c$|tkp|dkS)Nz...)Ellipsis)items r is_ellipsisr*/s 8  ,tu},r cdt|D}t|dkrt|d|dt|dkr|dSdS)Nc6g|]\}}t||Sr)r*)rirs r rz)single_ellipsis_index..4s)OOOga[=N=NOOOOr z1: More than one Ellipsis ('...') found in names (z-). This function supports up to one Ellipsis.r)rrr)rfn_nameellipsis_indicess r single_ellipsis_indexr23sOO5)9)9OOO !! D D D D D    !!"" 4r c6||t||z Sr)r)numel_pre_globnumel_post_globrs r expand_single_ellipsisr6?s #e**">> ??r ct|t||z dz |}|d||z||dzdzS)Nr/)r6r) ellipsis_idxr tensor_names globbed_namess r replace_ellipsis_by_positionr;CsS*c%jj</!3\M , - /% q8H8J8J2K KKr cLt||}||St|||S)zX Expands ... inside `names` to be equal to a list of names from `tensor_names`. )r2r;)rr9r0r8s r resolve_ellipsisr=Js0)88L ' e\ J JJr c t|dkr|d|d|S|t||jt ||S)Nr/r)r _update_namesr=rr&)rrr%s r update_names_with_listr@Tsc 5zzQ58+##D'222    nW.E.EFF  r c ^t|}|D]U}||}||vr|||<&tt|d|d|d|jd|d |t ||S)Nz: Tried to rename dim 'z ' to dim z in Tensor[z ] but dim 'z' does not exist)rkeysrr&rr?rvalues)r rename_mapr%dim_mapold_dimnew_dims r update_names_with_mappingrH^sF##G??$$W% gllnn $ $&GG  !'**ZZ7ZZZZ'-|ZZ@GZZZ    gnn&6&6 7 7 A AAr c >t|dk}t|}|rA|r?tt|dt|dt|d|s|st |||S|rt |||St |||S)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@rH)rrrDr%rhas_rename_pairss r update_namesrLlsFE QIJ'' % g&&  >LW>U>U  '5g'>'>       >->%feW===>%feW=== $VZ A AAr N) collectionsrr rr!r&r*r2r6r;r=r@rHrLrr r rNs######      ---   @@@LLLKKK B B B3B3B3B3B3Br