242242 < div class ="pytorch-left-menu-search ">
243243
244244 < div class ="version ">
245- < a href ='https://pytorch.org/docs/versions.html '> main (2.3.0a0+git55483fc ) ▼</ a >
245+ < a href ='https://pytorch.org/docs/versions.html '> main (2.3.0a0+gitb51e024 ) ▼</ a >
246246 </ div >
247247
248248
@@ -1129,7 +1129,7 @@ <h1>Source code for torch._lobpcg</h1><div class="highlight"><pre>
11291129 < span class ="n "> bparams</ span > < span class ="p "> [</ span > < span class ="s2 "> "ortho_use_drop"</ span > < span class ="p "> ]</ span > < span class ="o "> =</ span > < span class ="n "> bparams</ span > < span class ="o "> .</ span > < span class ="n "> get</ span > < span class ="p "> (</ span > < span class ="s2 "> "ortho_use_drop"</ span > < span class ="p "> ,</ span > < span class ="kc "> False</ span > < span class ="p "> )</ span >
11301130
11311131 < span class ="k "> if</ span > < span class ="ow "> not</ span > < span class ="n "> torch</ span > < span class ="o "> .</ span > < span class ="n "> jit</ span > < span class ="o "> .</ span > < span class ="n "> is_scripting</ span > < span class ="p "> ():</ span >
1132- < span class ="n "> LOBPCG</ span > < span class ="o "> .</ span > < span class ="n "> call_tracker</ span > < span class ="o "> =</ span > < span class ="n "> LOBPCG_call_tracker</ span > < span class ="c1 "> # type: ignore[assignment ]</ span >
1132+ < span class ="n "> LOBPCG</ span > < span class ="o "> .</ span > < span class ="n "> call_tracker</ span > < span class ="o "> =</ span > < span class ="n "> LOBPCG_call_tracker</ span > < span class ="c1 "> # type: ignore[method-assign ]</ span >
11331133
11341134 < span class ="k "> if</ span > < span class ="nb "> len</ span > < span class ="p "> (</ span > < span class ="n "> A</ span > < span class ="o "> .</ span > < span class ="n "> shape</ span > < span class ="p "> )</ span > < span class ="o "> ></ span > < span class ="mi "> 2</ span > < span class ="p "> :</ span >
11351135 < span class ="n "> N</ span > < span class ="o "> =</ span > < span class ="nb "> int</ span > < span class ="p "> (</ span > < span class ="n "> torch</ span > < span class ="o "> .</ span > < span class ="n "> prod</ span > < span class ="p "> (</ span > < span class ="n "> torch</ span > < span class ="o "> .</ span > < span class ="n "> tensor</ span > < span class ="p "> (</ span > < span class ="n "> A</ span > < span class ="o "> .</ span > < span class ="n "> shape</ span > < span class ="p "> [:</ span > < span class ="o "> -</ span > < span class ="mi "> 2</ span > < span class ="p "> ])))</ span >
@@ -1153,7 +1153,7 @@ <h1>Source code for torch._lobpcg</h1><div class="highlight"><pre>
11531153 < span class ="n "> bXret</ span > < span class ="p "> [</ span > < span class ="n "> i</ span > < span class ="p "> ]</ span > < span class ="o "> =</ span > < span class ="n "> worker</ span > < span class ="o "> .</ span > < span class ="n "> X</ span > < span class ="p "> [:,</ span > < span class ="p "> :</ span > < span class ="n "> k</ span > < span class ="p "> ]</ span >
11541154
11551155 < span class ="k "> if</ span > < span class ="ow "> not</ span > < span class ="n "> torch</ span > < span class ="o "> .</ span > < span class ="n "> jit</ span > < span class ="o "> .</ span > < span class ="n "> is_scripting</ span > < span class ="p "> ():</ span >
1156- < span class ="n "> LOBPCG</ span > < span class ="o "> .</ span > < span class ="n "> call_tracker</ span > < span class ="o "> =</ span > < span class ="n "> LOBPCG_call_tracker_orig</ span > < span class ="c1 "> # type: ignore[assignment ]</ span >
1156+ < span class ="n "> LOBPCG</ span > < span class ="o "> .</ span > < span class ="n "> call_tracker</ span > < span class ="o "> =</ span > < span class ="n "> LOBPCG_call_tracker_orig</ span > < span class ="c1 "> # type: ignore[method-assign ]</ span >
11571157
11581158 < span class ="k "> return</ span > < span class ="n "> bE</ span > < span class ="o "> .</ span > < span class ="n "> reshape</ span > < span class ="p "> (</ span > < span class ="n "> A</ span > < span class ="o "> .</ span > < span class ="n "> shape</ span > < span class ="p "> [:</ span > < span class ="o "> -</ span > < span class ="mi "> 2</ span > < span class ="p "> ]</ span > < span class ="o "> +</ span > < span class ="p "> (</ span > < span class ="n "> k</ span > < span class ="p "> ,)),</ span > < span class ="n "> bXret</ span > < span class ="o "> .</ span > < span class ="n "> reshape</ span > < span class ="p "> (</ span > < span class ="n "> A</ span > < span class ="o "> .</ span > < span class ="n "> shape</ span > < span class ="p "> [:</ span > < span class ="o "> -</ span > < span class ="mi "> 2</ span > < span class ="p "> ]</ span > < span class ="o "> +</ span > < span class ="p "> (</ span > < span class ="n "> m</ span > < span class ="p "> ,</ span > < span class ="n "> k</ span > < span class ="p "> ))</ span >
11591159
@@ -1165,7 +1165,7 @@ <h1>Source code for torch._lobpcg</h1><div class="highlight"><pre>
11651165 < span class ="n "> worker</ span > < span class ="o "> .</ span > < span class ="n "> run</ span > < span class ="p "> ()</ span >
11661166
11671167 < span class ="k "> if</ span > < span class ="ow "> not</ span > < span class ="n "> torch</ span > < span class ="o "> .</ span > < span class ="n "> jit</ span > < span class ="o "> .</ span > < span class ="n "> is_scripting</ span > < span class ="p "> ():</ span >
1168- < span class ="n "> LOBPCG</ span > < span class ="o "> .</ span > < span class ="n "> call_tracker</ span > < span class ="o "> =</ span > < span class ="n "> LOBPCG_call_tracker_orig</ span > < span class ="c1 "> # type: ignore[assignment ]</ span >
1168+ < span class ="n "> LOBPCG</ span > < span class ="o "> .</ span > < span class ="n "> call_tracker</ span > < span class ="o "> =</ span > < span class ="n "> LOBPCG_call_tracker_orig</ span > < span class ="c1 "> # type: ignore[method-assign ]</ span >
11691169
11701170 < span class ="k "> return</ span > < span class ="n "> worker</ span > < span class ="o "> .</ span > < span class ="n "> E</ span > < span class ="p "> [:</ span > < span class ="n "> k</ span > < span class ="p "> ],</ span > < span class ="n "> worker</ span > < span class ="o "> .</ span > < span class ="n "> X</ span > < span class ="p "> [:,</ span > < span class ="p "> :</ span > < span class ="n "> k</ span > < span class ="p "> ]</ span >
11711171
0 commit comments