192
192
< div class ="pytorch-left-menu-search ">
193
193
194
194
< 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 >
196
196
</ div >
197
197
198
198
@@ -1139,7 +1139,7 @@ <h1>Source code for torch._tensor</h1><div class="highlight"><pre>
1139
1139
1140
1140
< 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 >
1141
1141
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 >
1143
1143
< span class ="sa "> r</ span > < span class ="sd "> """Refines the dimension names of :attr:`self` according to :attr:`names`.</ span >
1144
1144
1145
1145
< 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>
1181
1181
< 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 >
1182
1182
< 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 >
1183
1183
< 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 >
1185
1185
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 >
1187
1187
< span class ="sa "> r</ span > < span class ="sd "> """Permutes the dimensions of the :attr:`self` tensor to match the order</ span >
1188
1188
< span class ="sd "> specified in :attr:`names`, adding size-one dims for any new names.</ span >
1189
1189
@@ -1226,9 +1226,9 @@ <h1>Source code for torch._tensor</h1><div class="highlight"><pre>
1226
1226
< 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 >
1227
1227
< 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 >
1228
1228
< 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 >
1230
1230
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 >
1232
1232
< span class ="sa "> r</ span > < span class ="sd "> """Expands the dimension :attr:`dim` of the :attr:`self` tensor over multiple dimensions</ span >
1233
1233
< span class ="sd "> of sizes given by :attr:`sizes`.</ span >
1234
1234
@@ -1268,10 +1268,10 @@ <h1>Source code for torch._tensor</h1><div class="highlight"><pre>
1268
1268
< span class ="n "> names</ span > < span class ="o "> =</ span > < span class ="kc "> None</ span >
1269
1269
< 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 >
1270
1270
< 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 >
1272
1272
1273
1273
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 >
1275
1275
< span class ="sd "> """In-place version of :meth:`~Tensor.rename`."""</ span >
1276
1276
1277
1277
< 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>
1282
1282
< span class ="c1 "> # 1) tensor.rename(*names) takes a vararglist of names</ span >
1283
1283
< span class ="c1 "> # 2) tensor.rename(**rename_map) takes a map of names to rename.</ span >
1284
1284
< 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 >
1286
1286
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 >
1288
1288
< span class ="sd "> """Renames dimension names of :attr:`self`.</ span >
1289
1289
1290
1290
< span class ="sd "> There are two main usages:</ span >
@@ -1322,7 +1322,7 @@ <h1>Source code for torch._tensor</h1><div class="highlight"><pre>
1322
1322
< 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 >
1323
1323
1324
1324
< 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 >
1326
1326
1327
1327
< 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 >
1328
1328
< span class ="sd "> """ Convert a tensor to compressed row storage format. Only works with 2D tensors.</ span >
0 commit comments