U ʹj@spddlmZddZddZddZdd Zd d Zd d ZddZddZ ddZ ddZ ddZ ddZ dS)) OrderedDictcCs|rtddS)NzzNYI: Named tensors don't support serialization. Please drop names via `tensor = tensor.rename(None)` before serialization.) has_names RuntimeErrortensorr[/var/www/html/bwave/python/venv/lib/python3.8/site-packages/torch/_namedtensor_internals.pycheck_serializing_named_tensor sr cCstddt|jDS)zkReturns a map of { dim: dim_name } where dim is a name if the dim is named and the dim index otherwise.cSs$g|]\}}|dkr|n||fqSNr).0idxnamerrr sz!build_dim_map..)r enumeratenamesrrrr build_dim_mapsrcCsTt|tr|}t|ds8t|ts8tdt|t|dkrLtdt|S)N__iter__zBExpected namedshape to be OrderedDict or iterable of tuples, got: rz!Expected namedshape to non-empty.) isinstanceritemshasattrtuplertypelenzip)Z namedshaperrrunzip_namedshapes   rcCs|rdSdSdS)NZrename_renamer)inplacerrrnamer_api_name&srcCs|tkp|dkS)Nz...)Ellipsis)itemrrr is_ellipsis-sr cCsJddt|D}t|dkr2t|d|dt|dkrF|dSdS)NcSsg|]\}}t|r|qSr)r )r ir rrrr2sz)single_ellipsis_index..z1: More than one Ellipsis ('...') found in names (z-). This function supports up to one Ellipsis.r)rrr)rfn_nameZellipsis_indicesrrrsingle_ellipsis_index1s  r%cCs||t||Sr )r)Znumel_pre_globZnumel_post_globrrrrexpand_single_ellipsis=sr&cCs8t|t||d|}|d||||ddS)Nr#)r&r) ellipsis_idxr tensor_namesZ globbed_namesrrrreplace_ellipsis_by_positionAs r)cCs"t||}|dkr|St|||S)zX Expands ... inside `names` to be equal to a list of names from `tensor_names`. N)r%r))rr(r$r'rrrresolve_ellipsisHs r*cCs>t|dkr$|ddkr$|d|S|t||jt||S)Nr#r)r _update_namesr*rr)rrrrrrupdate_names_with_listRs  r,c Cstt|}|D]N}||}||kr2|||<qtt|d|d|d|jd|d q|t||S)Nz: Tried to rename dim 'z ' to dim z in Tensor[z ] but dim 'z' does not exist)rkeysrrrr+rvalues)r rename_maprZdim_mapZold_dimZnew_dimrrrupdate_names_with_mapping\s   &r0cCsrt|dk}t|}|rB|rBtt|dt|dt|d|sV|sVt|||S|rft|||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.)rboolrrr,r0)rrr/rrZhas_rename_pairsrrr update_namesjs#    r2N) collectionsrr rrrr r%r&r)r*r,r0r2rrrrs