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+git7e69745 ) ▼</ a >
195+ < a href ='https://pytorch.org/docs/versions.html '> master (1.10.0a0+git4acea95 ) ▼</ a >
196196 </ div >
197197
198198
@@ -1139,7 +1139,7 @@ <h1>Source code for torch._tensor</h1><div class="highlight"><pre>
11391139
11401140 < 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 >
11411141
1142- < 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 >
1142+ < 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 >
11431143 < span class ="sa "> r</ span > < span class ="sd "> """Refines the dimension names of :attr:`self` according to :attr:`names`.</ span >
11441144
11451145< span class ="sd "> Refining is a special case of renaming that "lifts" unnamed dimensions.</ span >
@@ -1181,9 +1181,9 @@ <h1>Source code for torch._tensor</h1><div class="highlight"><pre>
11811181 < 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 >
11821182 < 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 >
11831183 < 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 >
1184- < 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 >
1184+ < 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 >
11851185
1186- < 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 >
1186+ < 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 >
11871187 < span class ="sa "> r</ span > < span class ="sd "> """Permutes the dimensions of the :attr:`self` tensor to match the order</ span >
11881188< span class ="sd "> specified in :attr:`names`, adding size-one dims for any new names.</ span >
11891189
@@ -1226,9 +1226,9 @@ <h1>Source code for torch._tensor</h1><div class="highlight"><pre>
12261226 < 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 >
12271227 < 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 >
12281228 < 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 >
1229- < span class ="n "> ellipsis_idx</ span > < span class ="p "> )</ span > </ div >
1229+ < span class ="n "> ellipsis_idx</ span > < span class ="p "> )</ span >
12301230
1231- < 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 >
1231+ < 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 >
12321232 < span class ="sa "> r</ span > < span class ="sd "> """Expands the dimension :attr:`dim` of the :attr:`self` tensor over multiple dimensions</ span >
12331233< span class ="sd "> of sizes given by :attr:`sizes`.</ span >
12341234
@@ -1268,10 +1268,10 @@ <h1>Source code for torch._tensor</h1><div class="highlight"><pre>
12681268 < span class ="n "> names</ span > < span class ="o "> =</ span > < span class ="kc "> None</ span >
12691269 < 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 >
12701270 < 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 >
1271- < 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 >
1271+ < 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 >
12721272
12731273
1274- < 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 >
1274+ < 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 >
12751275 < span class ="sd "> """In-place version of :meth:`~Tensor.rename`."""</ span >
12761276
12771277 < 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 >
@@ -1282,9 +1282,9 @@ <h1>Source code for torch._tensor</h1><div class="highlight"><pre>
12821282 < span class ="c1 "> # 1) tensor.rename(*names) takes a vararglist of names</ span >
12831283 < span class ="c1 "> # 2) tensor.rename(**rename_map) takes a map of names to rename.</ span >
12841284 < span class ="c1 "> # C++ is static, making it difficult to implement similar behavior.</ span >
1285- < 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 >
1285+ < 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 >
12861286
1287- < 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 >
1287+ < 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 >
12881288 < span class ="sd "> """Renames dimension names of :attr:`self`.</ span >
12891289
12901290< span class ="sd "> There are two main usages:</ span >
@@ -1322,7 +1322,7 @@ <h1>Source code for torch._tensor</h1><div class="highlight"><pre>
13221322 < 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 >
13231323
13241324 < span class ="c1 "> # See Note [rename_ / rename API]</ span >
1325- < 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 >
1325+ < 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 >
13261326
13271327 < 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 >
13281328 < span class ="sd "> """ Convert a tensor to compressed row storage format. Only works with 2D tensors.</ span >
0 commit comments