213213 < div class ="pytorch-left-menu-search ">
214214
215215 < 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 >
217217 </ div >
218218
219219
@@ -1069,17 +1069,16 @@ <h1>Source code for torch._lobpcg</h1><div class="highlight"><pre>
10691069< span class ="sd "> """</ span >
10701070
10711071 < 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 >
10831082
10841083 < span class ="c1 "> # constant parameters</ span >
10851084 < 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>
10981097 < 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 >
10991098 < 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 >
11001099 < 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 >
11051104
11061105 < span class ="k "> def</ span > < span class ="fm "> __str__</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="p "> ):</ span >
11071106 < 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>
13641363 < span class ="k "> return</ span > < span class ="n "> Rinv</ span > < span class ="o "> *</ span > < span class ="n "> d_col</ span >
13651364
13661365 < 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 >
13721370 < span class ="sd "> """Return B-orthonormal U.</ span >
13731371
13741372< span class ="sd "> .. note:: When `drop` is `False` then `svqb` is based on the</ span >
0 commit comments