213
213
< div class ="pytorch-left-menu-search ">
214
214
215
215
< div class ="version ">
216
- < a href ='https://pytorch.org/docs/versions.html '> master (1.12.0a0+git943085f ) ▼</ a >
216
+ < a href ='https://pytorch.org/docs/versions.html '> master (1.12.0a0+git1c152f8 ) ▼</ a >
217
217
</ div >
218
218
219
219
@@ -1069,17 +1069,16 @@ <h1>Source code for torch._lobpcg</h1><div class="highlight"><pre>
1069
1069
< span class ="sd "> """</ span >
1070
1070
1071
1071
< span class ="k "> def</ span > < span class ="fm "> __init__</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="p "> ,</ span >
1072
- < span class ="n "> A</ span > < span class ="p "> ,</ span > < span class ="c1 "> # type: Optional[Tensor]</ span >
1073
- < span class ="n "> B</ span > < span class ="p "> ,</ span > < span class ="c1 "> # type: Optional[Tensor]</ span >
1074
- < span class ="n "> X</ span > < span class ="p "> ,</ span > < span class ="c1 "> # type: Tensor</ span >
1075
- < span class ="n "> iK</ span > < span class ="p "> ,</ span > < span class ="c1 "> # type: Optional[Tensor]</ span >
1076
- < span class ="n "> iparams</ span > < span class ="p "> ,</ span > < span class ="c1 "> # type: Dict[str, int]</ span >
1077
- < span class ="n "> fparams</ span > < span class ="p "> ,</ span > < span class ="c1 "> # type: Dict[str, float]</ span >
1078
- < span class ="n "> bparams</ span > < span class ="p "> ,</ span > < span class ="c1 "> # type: Dict[str, bool]</ span >
1079
- < span class ="n "> method</ span > < span class ="p "> ,</ span > < span class ="c1 "> # type: str</ span >
1080
- < span class ="n "> tracker</ span > < span class ="c1 "> # type: None</ span >
1081
- < span class ="p "> ):</ span >
1082
- < span class ="c1 "> # type: (...) -> None</ span >
1072
+ < span class ="n "> A</ span > < span class ="p "> :</ span > < span class ="n "> Optional</ span > < span class ="p "> [</ span > < span class ="n "> Tensor</ span > < span class ="p "> ],</ span >
1073
+ < span class ="n "> B</ span > < span class ="p "> :</ span > < span class ="n "> Optional</ span > < span class ="p "> [</ span > < span class ="n "> Tensor</ span > < span class ="p "> ],</ span >
1074
+ < span class ="n "> X</ span > < span class ="p "> :</ span > < span class ="n "> Tensor</ span > < span class ="p "> ,</ span >
1075
+ < span class ="n "> iK</ span > < span class ="p "> :</ span > < span class ="n "> Optional</ span > < span class ="p "> [</ span > < span class ="n "> Tensor</ span > < span class ="p "> ],</ span >
1076
+ < span class ="n "> iparams</ span > < span class ="p "> :</ span > < span class ="n "> Dict</ span > < span class ="p "> [</ span > < span class ="nb "> str</ span > < span class ="p "> ,</ span > < span class ="nb "> int</ span > < span class ="p "> ],</ span >
1077
+ < span class ="n "> fparams</ span > < span class ="p "> :</ span > < span class ="n "> Dict</ span > < span class ="p "> [</ span > < span class ="nb "> str</ span > < span class ="p "> ,</ span > < span class ="nb "> float</ span > < span class ="p "> ],</ span >
1078
+ < span class ="n "> bparams</ span > < span class ="p "> :</ span > < span class ="n "> Dict</ span > < span class ="p "> [</ span > < span class ="nb "> str</ span > < span class ="p "> ,</ span > < span class ="nb "> bool</ span > < span class ="p "> ],</ span >
1079
+ < span class ="n "> method</ span > < span class ="p "> :</ span > < span class ="nb "> str</ span > < span class ="p "> ,</ span >
1080
+ < span class ="n "> tracker</ span > < span class ="p "> :</ span > < span class ="kc "> None</ span >
1081
+ < span class ="p "> )</ span > < span class ="o "> -></ span > < span class ="kc "> None</ span > < span class ="p "> :</ span >
1083
1082
1084
1083
< span class ="c1 "> # constant parameters</ span >
1085
1084
< span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> A</ span > < span class ="o "> =</ span > < span class ="n "> A</ span >
@@ -1098,10 +1097,10 @@ <h1>Source code for torch._lobpcg</h1><div class="highlight"><pre>
1098
1097
< span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> E</ span > < span class ="o "> =</ span > < span class ="n "> torch</ span > < span class ="o "> .</ span > < span class ="n "> zeros</ span > < span class ="p "> ((</ span > < span class ="n "> n</ span > < span class ="p "> ,</ span > < span class ="p "> ),</ span > < span class ="n "> dtype</ span > < span class ="o "> =</ span > < span class ="n "> X</ span > < span class ="o "> .</ span > < span class ="n "> dtype</ span > < span class ="p "> ,</ span > < span class ="n "> device</ span > < span class ="o "> =</ span > < span class ="n "> X</ span > < span class ="o "> .</ span > < span class ="n "> device</ span > < span class ="p "> )</ span >
1099
1098
< span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> R</ span > < span class ="o "> =</ span > < span class ="n "> torch</ span > < span class ="o "> .</ span > < span class ="n "> zeros</ span > < span class ="p "> ((</ span > < span class ="n "> m</ span > < span class ="p "> ,</ span > < span class ="n "> n</ span > < span class ="p "> ),</ span > < span class ="n "> dtype</ span > < span class ="o "> =</ span > < span class ="n "> X</ span > < span class ="o "> .</ span > < span class ="n "> dtype</ span > < span class ="p "> ,</ span > < span class ="n "> device</ span > < span class ="o "> =</ span > < span class ="n "> X</ span > < span class ="o "> .</ span > < span class ="n "> device</ span > < span class ="p "> )</ span >
1100
1099
< span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> S</ span > < span class ="o "> =</ span > < span class ="n "> torch</ span > < span class ="o "> .</ span > < span class ="n "> zeros</ span > < span class ="p "> ((</ span > < span class ="n "> m</ span > < span class ="p "> ,</ span > < span class ="mi "> 3</ span > < span class ="o "> *</ span > < span class ="n "> n</ span > < span class ="p "> ),</ span > < span class ="n "> dtype</ span > < span class ="o "> =</ span > < span class ="n "> X</ span > < span class ="o "> .</ span > < span class ="n "> dtype</ span > < span class ="p "> ,</ span > < span class ="n "> device</ span > < span class ="o "> =</ span > < span class ="n "> X</ span > < span class ="o "> .</ span > < span class ="n "> device</ span > < span class ="p "> )</ span >
1101
- < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> tvars</ span > < span class ="o " > = </ span > < span class ="p "> {} </ span > < span class ="c1 " > # type: Dict[str, Tensor] </ span >
1102
- < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> ivars</ span > < span class ="o " > = </ span > < span class ="p "> { </ span > < span class ="s1 " > 'istep' </ span > < span class ="p "> : </ span > < span class ="mi " > 0 </ span > < span class ="p "> } </ span > < span class ="c1 " > # type: Dict[str, int] </ span >
1103
- < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> fvars</ span > < span class ="o " > = </ span > < span class ="p "> { </ span > < span class ="s1 " > '_' </ span > < span class ="p "> : </ span > < span class ="mf " > 0.0 </ span > < span class ="p "> } </ span > < span class ="c1 " > # type: Dict[str, float] </ span >
1104
- < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> bvars</ span > < span class ="o " > = </ span > < span class ="p "> { </ span > < span class ="s1 " > '_' </ span > < span class ="p "> : </ span > < span class ="kc " > False </ span > < span class ="p "> } </ span > < span class ="c1 " > # type: Dict[str, bool] </ span >
1100
+ < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> tvars</ span > < span class =" p " > : </ span > < span class ="n " > Dict </ span > < span class =" p " > [ </ span > < span class =" nb " > str </ span > < span class ="p "> , </ span > < span class =" n " > Tensor </ span > < span class =" p " > ] </ span > < span class ="o " > = </ span > < span class =" p " > {} </ span >
1101
+ < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> ivars</ span > < span class ="p " > : </ span > < span class ="n " > Dict </ span > < span class =" p "> [ </ span > < span class ="nb " > str </ span > < span class ="p "> , </ span > < span class ="nb " > int </ span > < span class ="p "> ] </ span > < span class =" o " > = </ span > < span class =" p " > { </ span > < span class ="s1 " > 'istep' </ span > < span class =" p " > : </ span > < span class =" mi " > 0 </ span > < span class =" p " > } </ span >
1102
+ < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> fvars</ span > < span class ="p " > : </ span > < span class ="n " > Dict </ span > < span class =" p "> [ </ span > < span class ="nb " > str </ span > < span class ="p "> , </ span > < span class ="nb " > float </ span > < span class ="p "> ] </ span > < span class =" o " > = </ span > < span class =" p " > { </ span > < span class ="s1 " > '_' </ span > < span class =" p " > : </ span > < span class =" mf " > 0.0 </ span > < span class =" p " > } </ span >
1103
+ < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> bvars</ span > < span class ="p " > : </ span > < span class ="n " > Dict </ span > < span class =" p "> [ </ span > < span class ="nb " > str </ span > < span class ="p "> , </ span > < span class ="nb " > bool </ span > < span class ="p "> ] </ span > < span class =" o " > = </ span > < span class =" p " > { </ span > < span class ="s1 " > '_' </ span > < span class =" p " > : </ span > < span class =" kc " > False </ span > < span class =" p " > } </ span >
1105
1104
1106
1105
< span class ="k "> def</ span > < span class ="fm "> __str__</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="p "> ):</ span >
1107
1106
< span class ="n "> lines</ span > < span class ="o "> =</ span > < span class ="p "> [</ span > < span class ="s1 "> 'LOPBCG:'</ span > < span class ="p "> ]</ span >
@@ -1364,11 +1363,10 @@ <h1>Source code for torch._lobpcg</h1><div class="highlight"><pre>
1364
1363
< span class ="k "> return</ span > < span class ="n "> Rinv</ span > < span class ="o "> *</ span > < span class ="n "> d_col</ span >
1365
1364
1366
1365
< span class ="k "> def</ span > < span class ="nf "> _get_svqb</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="p "> ,</ span >
1367
- < span class ="n "> U</ span > < span class ="p "> ,</ span > < span class ="c1 "> # Tensor</ span >
1368
- < span class ="n "> drop</ span > < span class ="p "> ,</ span > < span class ="c1 "> # bool</ span >
1369
- < span class ="n "> tau</ span > < span class ="c1 "> # float</ span >
1370
- < span class ="p "> ):</ span >
1371
- < span class ="c1 "> # type: (Tensor, bool, float) -> Tensor</ span >
1366
+ < span class ="n "> U</ span > < span class ="p "> :</ span > < span class ="n "> Tensor</ span > < span class ="p "> ,</ span > < span class ="c1 "> # Tensor</ span >
1367
+ < span class ="n "> drop</ span > < span class ="p "> :</ span > < span class ="nb "> bool</ span > < span class ="p "> ,</ span > < span class ="c1 "> # bool</ span >
1368
+ < span class ="n "> tau</ span > < span class ="p "> :</ span > < span class ="nb "> float</ span > < span class ="c1 "> # float</ span >
1369
+ < span class ="p "> )</ span > < span class ="o "> -></ span > < span class ="n "> Tensor</ span > < span class ="p "> :</ span >
1372
1370
< span class ="sd "> """Return B-orthonormal U.</ span >
1373
1371
1374
1372
< span class ="sd "> .. note:: When `drop` is `False` then `svqb` is based on the</ span >
0 commit comments