192192 < div class ="pytorch-left-menu-search ">
193193
194194 < div class ="version ">
195- < a href ='https://pytorch.org/docs/versions.html '> master (1.10.0a0+gitb324086 ) ▼</ a >
195+ < a href ='https://pytorch.org/docs/versions.html '> master (1.10.0a0+gitcf42c96 ) ▼</ a >
196196 </ div >
197197
198198
@@ -1129,7 +1129,7 @@ <h1>Source code for torch._tensor</h1><div class="highlight"><pre>
11291129
11301130 < span class ="k "> return</ span > < span class ="nb "> dict</ span > < span class ="p "> (</ span > < span class ="n "> typestr</ span > < span class ="o "> =</ span > < span class ="n "> typestr</ span > < span class ="p "> ,</ span > < span class ="n "> shape</ span > < span class ="o "> =</ span > < span class ="n "> shape</ span > < span class ="p "> ,</ span > < span class ="n "> strides</ span > < span class ="o "> =</ span > < span class ="n "> strides</ span > < span class ="p "> ,</ span > < span class ="n "> data</ span > < span class ="o "> =</ span > < span class ="n "> data</ span > < span class ="p "> ,</ span > < span class ="n "> version</ span > < span class ="o "> =</ span > < span class ="mi "> 2</ span > < span class ="p "> )</ span >
11311131
1132- < div class =" viewcode-block " id =" Tensor.refine_names " > < a class =" viewcode-back " href =" ../../named_tensor.html#torch.Tensor.refine_names " > [docs] </ a > < span class ="k "> def</ span > < span class ="nf "> refine_names</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="p "> ,</ span > < span class ="o "> *</ span > < span class ="n "> names</ span > < span class ="p "> ):</ span >
1132+ < span class ="k "> def</ span > < span class ="nf "> refine_names</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="p "> ,</ span > < span class ="o "> *</ span > < span class ="n "> names</ span > < span class ="p "> ):</ span >
11331133 < span class ="sa "> r</ span > < span class ="sd "> """Refines the dimension names of :attr:`self` according to :attr:`names`.</ span >
11341134
11351135< span class ="sd "> Refining is a special case of renaming that "lifts" unnamed dimensions.</ span >
@@ -1171,9 +1171,9 @@ <h1>Source code for torch._tensor</h1><div class="highlight"><pre>
11711171 < span class ="k "> if</ span > < span class ="n "> has_torch_function_unary</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="p "> ):</ span >
11721172 < span class ="k "> return</ span > < span class ="n "> handle_torch_function</ span > < span class ="p "> (</ span > < span class ="n "> Tensor</ span > < span class ="o "> .</ span > < span class ="n "> refine_names</ span > < span class ="p "> ,</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="p "> ,),</ span > < span class ="bp "> self</ span > < span class ="p "> ,</ span > < span class ="o "> *</ span > < span class ="n "> names</ span > < span class ="p "> )</ span >
11731173 < span class ="n "> names</ span > < span class ="o "> =</ span > < span class ="n "> resolve_ellipsis</ span > < span class ="p "> (</ span > < span class ="n "> names</ span > < span class ="p "> ,</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> names</ span > < span class ="p "> ,</ span > < span class ="s1 "> 'refine_names'</ span > < span class ="p "> )</ span >
1174- < span class ="k "> return</ span > < span class ="nb "> super</ span > < span class ="p "> (</ span > < span class ="n "> Tensor</ span > < span class ="p "> ,</ span > < span class ="bp "> self</ span > < span class ="p "> )</ span > < span class ="o "> .</ span > < span class ="n "> refine_names</ span > < span class ="p "> (</ span > < span class ="n "> names</ span > < span class ="p "> )</ span > </ div >
1174+ < span class ="k "> return</ span > < span class ="nb "> super</ span > < span class ="p "> (</ span > < span class ="n "> Tensor</ span > < span class ="p "> ,</ span > < span class ="bp "> self</ span > < span class ="p "> )</ span > < span class ="o "> .</ span > < span class ="n "> refine_names</ span > < span class ="p "> (</ span > < span class ="n "> names</ span > < span class ="p "> )</ span >
11751175
1176- < div class =" viewcode-block " id =" Tensor.align_to " > < a class =" viewcode-back " href =" ../../named_tensor.html#torch.Tensor.align_to " > [docs] </ a > < span class ="k "> def</ span > < span class ="nf "> align_to</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="p "> ,</ span > < span class ="o "> *</ span > < span class ="n "> names</ span > < span class ="p "> ):</ span >
1176+ < span class ="k "> def</ span > < span class ="nf "> align_to</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="p "> ,</ span > < span class ="o "> *</ span > < span class ="n "> names</ span > < span class ="p "> ):</ span >
11771177 < span class ="sa "> r</ span > < span class ="sd "> """Permutes the dimensions of the :attr:`self` tensor to match the order</ span >
11781178< span class ="sd "> specified in :attr:`names`, adding size-one dims for any new names.</ span >
11791179
@@ -1216,9 +1216,9 @@ <h1>Source code for torch._tensor</h1><div class="highlight"><pre>
12161216 < span class ="k "> return</ span > < span class ="nb "> super</ span > < span class ="p "> (</ span > < span class ="n "> Tensor</ span > < span class ="p "> ,</ span > < span class ="bp "> self</ span > < span class ="p "> )</ span > < span class ="o "> .</ span > < span class ="n "> align_to</ span > < span class ="p "> (</ span > < span class ="n "> names</ span > < span class ="p "> )</ span >
12171217 < span class ="k "> return</ span > < span class ="nb "> super</ span > < span class ="p "> (</ span > < span class ="n "> Tensor</ span > < span class ="p "> ,</ span > < span class ="bp "> self</ span > < span class ="p "> )</ span > < span class ="o "> .</ span > < span class ="n "> align_to</ span > < span class ="p "> (</ span >
12181218 < span class ="p "> [</ span > < span class ="n "> name</ span > < span class ="k "> for</ span > < span class ="n "> name</ span > < span class ="ow "> in</ span > < span class ="n "> names</ span > < span class ="k "> if</ span > < span class ="ow "> not</ span > < span class ="n "> is_ellipsis</ span > < span class ="p "> (</ span > < span class ="n "> name</ span > < span class ="p "> )],</ span >
1219- < span class ="n "> ellipsis_idx</ span > < span class ="p "> )</ span > </ div >
1219+ < span class ="n "> ellipsis_idx</ span > < span class ="p "> )</ span >
12201220
1221- < div class =" viewcode-block " id =" Tensor.unflatten " > < a class =" viewcode-back " href =" ../../named_tensor.html#torch.Tensor.unflatten " > [docs] </ a > < span class ="k "> def</ span > < span class ="nf "> unflatten</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="p "> ,</ span > < span class ="n "> dim</ span > < span class ="p "> ,</ span > < span class ="n "> sizes</ span > < span class ="p "> ):</ span >
1221+ < span class ="k "> def</ span > < span class ="nf "> unflatten</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="p "> ,</ span > < span class ="n "> dim</ span > < span class ="p "> ,</ span > < span class ="n "> sizes</ span > < span class ="p "> ):</ span >
12221222 < span class ="sa "> r</ span > < span class ="sd "> """Expands the dimension :attr:`dim` of the :attr:`self` tensor over multiple dimensions</ span >
12231223< span class ="sd "> of sizes given by :attr:`sizes`.</ span >
12241224
@@ -1258,10 +1258,10 @@ <h1>Source code for torch._tensor</h1><div class="highlight"><pre>
12581258 < span class ="n "> names</ span > < span class ="o "> =</ span > < span class ="kc "> None</ span >
12591259 < span class ="k "> if</ span > < span class ="nb "> isinstance</ span > < span class ="p "> (</ span > < span class ="n "> sizes</ span > < span class ="p "> ,</ span > < span class ="n "> OrderedDict</ span > < span class ="p "> )</ span > < span class ="ow "> or</ span > < span class ="p "> (</ span > < span class ="nb "> isinstance</ span > < span class ="p "> (</ span > < span class ="n "> sizes</ span > < span class ="p "> ,</ span > < span class ="p "> (</ span > < span class ="nb "> tuple</ span > < span class ="p "> ,</ span > < span class ="nb "> list</ span > < span class ="p "> ))</ span > < span class ="ow "> and</ span > < span class ="nb "> isinstance</ span > < span class ="p "> (</ span > < span class ="n "> sizes</ span > < span class ="p "> [</ span > < span class ="mi "> 0</ span > < span class ="p "> ],</ span > < span class ="p "> (</ span > < span class ="nb "> tuple</ span > < span class ="p "> ,</ span > < span class ="nb "> list</ span > < span class ="p "> ))):</ span >
12601260 < span class ="n "> names</ span > < span class ="p "> ,</ span > < span class ="n "> sizes</ span > < span class ="o "> =</ span > < span class ="n "> unzip_namedshape</ span > < span class ="p "> (</ span > < span class ="n "> sizes</ span > < span class ="p "> )</ span >
1261- < span class ="k "> return</ span > < span class ="nb "> super</ span > < span class ="p "> (</ span > < span class ="n "> Tensor</ span > < span class ="p "> ,</ span > < span class ="bp "> self</ span > < span class ="p "> )</ span > < span class ="o "> .</ span > < span class ="n "> unflatten</ span > < span class ="p "> (</ span > < span class ="n "> dim</ span > < span class ="p "> ,</ span > < span class ="n "> sizes</ span > < span class ="p "> ,</ span > < span class ="n "> names</ span > < span class ="p "> )</ span > </ div >
1261+ < span class ="k "> return</ span > < span class ="nb "> super</ span > < span class ="p "> (</ span > < span class ="n "> Tensor</ span > < span class ="p "> ,</ span > < span class ="bp "> self</ span > < span class ="p "> )</ span > < span class ="o "> .</ span > < span class ="n "> unflatten</ span > < span class ="p "> (</ span > < span class ="n "> dim</ span > < span class ="p "> ,</ span > < span class ="n "> sizes</ span > < span class ="p "> ,</ span > < span class ="n "> names</ span > < span class ="p "> )</ span >
12621262
12631263
1264- < div class =" viewcode-block " id =" Tensor.rename_ " > < a class =" viewcode-back " href =" ../../named_tensor.html#torch.Tensor.rename_ " > [docs] </ a > < span class ="k "> def</ span > < span class ="nf "> rename_</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="p "> ,</ span > < span class ="o "> *</ span > < span class ="n "> names</ span > < span class ="p "> ,</ span > < span class ="o "> **</ span > < span class ="n "> rename_map</ span > < span class ="p "> ):</ span >
1264+ < span class ="k "> def</ span > < span class ="nf "> rename_</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="p "> ,</ span > < span class ="o "> *</ span > < span class ="n "> names</ span > < span class ="p "> ,</ span > < span class ="o "> **</ span > < span class ="n "> rename_map</ span > < span class ="p "> ):</ span >
12651265 < span class ="sd "> """In-place version of :meth:`~Tensor.rename`."""</ span >
12661266
12671267 < span class ="k "> if</ span > < span class ="n "> has_torch_function_unary</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="p "> ):</ span >
@@ -1272,9 +1272,9 @@ <h1>Source code for torch._tensor</h1><div class="highlight"><pre>
12721272 < span class ="c1 "> # 1) tensor.rename(*names) takes a vararglist of names</ span >
12731273 < span class ="c1 "> # 2) tensor.rename(**rename_map) takes a map of names to rename.</ span >
12741274 < span class ="c1 "> # C++ is static, making it difficult to implement similar behavior.</ span >
1275- < span class ="k "> return</ span > < span class ="n "> update_names</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="p "> ,</ span > < span class ="n "> names</ span > < span class ="p "> ,</ span > < span class ="n "> rename_map</ span > < span class ="p "> ,</ span > < span class ="n "> inplace</ span > < span class ="o "> =</ span > < span class ="kc "> True</ span > < span class ="p "> )</ span > </ div >
1275+ < span class ="k "> return</ span > < span class ="n "> update_names</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="p "> ,</ span > < span class ="n "> names</ span > < span class ="p "> ,</ span > < span class ="n "> rename_map</ span > < span class ="p "> ,</ span > < span class ="n "> inplace</ span > < span class ="o "> =</ span > < span class ="kc "> True</ span > < span class ="p "> )</ span >
12761276
1277- < div class =" viewcode-block " id =" Tensor.rename " > < a class =" viewcode-back " href =" ../../named_tensor.html#torch.Tensor.rename " > [docs] </ a > < span class ="k "> def</ span > < span class ="nf "> rename</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="p "> ,</ span > < span class ="o "> *</ span > < span class ="n "> names</ span > < span class ="p "> ,</ span > < span class ="o "> **</ span > < span class ="n "> rename_map</ span > < span class ="p "> ):</ span >
1277+ < span class ="k "> def</ span > < span class ="nf "> rename</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="p "> ,</ span > < span class ="o "> *</ span > < span class ="n "> names</ span > < span class ="p "> ,</ span > < span class ="o "> **</ span > < span class ="n "> rename_map</ span > < span class ="p "> ):</ span >
12781278 < span class ="sd "> """Renames dimension names of :attr:`self`.</ span >
12791279
12801280< span class ="sd "> There are two main usages:</ span >
@@ -1312,7 +1312,7 @@ <h1>Source code for torch._tensor</h1><div class="highlight"><pre>
13121312 < span class ="k "> return</ span > < span class ="n "> handle_torch_function</ span > < span class ="p "> (</ span > < span class ="n "> Tensor</ span > < span class ="o "> .</ span > < span class ="n "> rename</ span > < span class ="p "> ,</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="p "> ,),</ span > < span class ="bp "> self</ span > < span class ="p "> ,</ span > < span class ="o "> *</ span > < span class ="n "> names</ span > < span class ="p "> ,</ span > < span class ="o "> **</ span > < span class ="n "> rename_map</ span > < span class ="p "> )</ span >
13131313
13141314 < span class ="c1 "> # See Note [rename_ / rename API]</ span >
1315- < span class ="k "> return</ span > < span class ="n "> update_names</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="p "> ,</ span > < span class ="n "> names</ span > < span class ="p "> ,</ span > < span class ="n "> rename_map</ span > < span class ="p "> ,</ span > < span class ="n "> inplace</ span > < span class ="o "> =</ span > < span class ="kc "> False</ span > < span class ="p "> )</ span > </ div >
1315+ < span class ="k "> return</ span > < span class ="n "> update_names</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="p "> ,</ span > < span class ="n "> names</ span > < span class ="p "> ,</ span > < span class ="n "> rename_map</ span > < span class ="p "> ,</ span > < span class ="n "> inplace</ span > < span class ="o "> =</ span > < span class ="kc "> False</ span > < span class ="p "> )</ span >
13161316
13171317 < span class ="k "> def</ span > < span class ="nf "> to_sparse_csr</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="p "> ):</ span >
13181318 < span class ="sd "> """ Convert a tensor to compressed row storage format. Only works with 2D tensors.</ span >
0 commit comments