@@ -488,7 +488,7 @@ <h1>Source code for torch._lowrank</h1><div class="highlight"><pre>
488
488
< span class ="k "> return</ span > < span class ="n "> Q</ span >
489
489
490
490
491
- < div class =" viewcode-block " id =" svd_lowrank " > < a class =" viewcode-back " href =" ../../generated/torch.svd_lowrank.html#torch.svd_lowrank " > [docs] </ a > < span class ="k "> def</ span > < span class ="nf "> svd_lowrank</ span > < span class ="p "> (</ span > < span class ="n "> A</ span > < span class ="p "> :</ span > < span class ="n "> Tensor</ span > < span class ="p "> ,</ span > < span class ="n "> q</ span > < span class ="p "> :</ span > < span class ="n "> Optional</ span > < span class ="p "> [</ span > < span class ="nb "> int</ span > < span class ="p "> ]</ span > < span class ="o "> =</ span > < span class ="mi "> 6</ span > < span class ="p "> ,</ span > < span class ="n "> niter</ span > < span class ="p "> :</ span > < span class ="n "> Optional</ span > < span class ="p "> [</ span > < span class ="nb "> int</ span > < span class ="p "> ]</ span > < span class ="o "> =</ span > < span class ="mi "> 2</ span > < span class ="p "> ,</ span >
491
+ < span class ="k "> def</ span > < span class ="nf "> svd_lowrank</ span > < span class ="p "> (</ span > < span class ="n "> A</ span > < span class ="p "> :</ span > < span class ="n "> Tensor</ span > < span class ="p "> ,</ span > < span class ="n "> q</ span > < span class ="p "> :</ span > < span class ="n "> Optional</ span > < span class ="p "> [</ span > < span class ="nb "> int</ span > < span class ="p "> ]</ span > < span class ="o "> =</ span > < span class ="mi "> 6</ span > < span class ="p "> ,</ span > < span class ="n "> niter</ span > < span class ="p "> :</ span > < span class ="n "> Optional</ span > < span class ="p "> [</ span > < span class ="nb "> int</ span > < span class ="p "> ]</ span > < span class ="o "> =</ span > < span class ="mi "> 2</ span > < span class ="p "> ,</ span >
492
492
< span class ="n "> M</ span > < span class ="p "> :</ span > < span class ="n "> Optional</ span > < span class ="p "> [</ span > < span class ="n "> Tensor</ span > < span class ="p "> ]</ span > < span class ="o "> =</ span > < span class ="kc "> None</ span > < span class ="p "> )</ span > < span class ="o "> -></ span > < span class ="n "> Tuple</ span > < span class ="p "> [</ span > < span class ="n "> Tensor</ span > < span class ="p "> ,</ span > < span class ="n "> Tensor</ span > < span class ="p "> ,</ span > < span class ="n "> Tensor</ span > < span class ="p "> ]:</ span >
493
493
< span class ="sa "> r</ span > < span class ="sd "> """Return the singular value decomposition ``(U, S, V)`` of a matrix,</ span >
494
494
< span class ="sd "> batches of matrices, or a sparse matrix :math:`A` such that</ span >
@@ -533,7 +533,7 @@ <h1>Source code for torch._lowrank</h1><div class="highlight"><pre>
533
533
< span class ="n "> tensor_ops</ span > < span class ="o "> =</ span > < span class ="p "> (</ span > < span class ="n "> A</ span > < span class ="p "> ,</ span > < span class ="n "> M</ span > < span class ="p "> )</ span >
534
534
< span class ="k "> if</ span > < span class ="p "> (</ span > < span class ="ow "> not</ span > < span class ="nb "> set</ span > < span class ="p "> (</ span > < span class ="nb "> map</ span > < span class ="p "> (</ span > < span class ="nb "> type</ span > < span class ="p "> ,</ span > < span class ="n "> tensor_ops</ span > < span class ="p "> ))</ span > < span class ="o "> .</ span > < span class ="n "> issubset</ 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 ="nb "> type</ span > < span class ="p "> (</ span > < span class ="kc "> None</ span > < span class ="p "> )))</ span > < span class ="ow "> and</ span > < span class ="n "> has_torch_function</ span > < span class ="p "> (</ span > < span class ="n "> tensor_ops</ span > < span class ="p "> )):</ span >
535
535
< span class ="k "> return</ span > < span class ="n "> handle_torch_function</ span > < span class ="p "> (</ span > < span class ="n "> svd_lowrank</ span > < span class ="p "> ,</ span > < span class ="n "> tensor_ops</ span > < span class ="p "> ,</ span > < span class ="n "> A</ span > < span class ="p "> ,</ span > < span class ="n "> q</ span > < span class ="o "> =</ span > < span class ="n "> q</ span > < span class ="p "> ,</ span > < span class ="n "> niter</ span > < span class ="o "> =</ span > < span class ="n "> niter</ span > < span class ="p "> ,</ span > < span class ="n "> M</ span > < span class ="o "> =</ span > < span class ="n "> M</ span > < span class ="p "> )</ span >
536
- < span class ="k "> return</ span > < span class ="n "> _svd_lowrank</ span > < span class ="p "> (</ span > < span class ="n "> A</ span > < span class ="p "> ,</ span > < span class ="n "> q</ span > < span class ="o "> =</ span > < span class ="n "> q</ span > < span class ="p "> ,</ span > < span class ="n "> niter</ span > < span class ="o "> =</ span > < span class ="n "> niter</ span > < span class ="p "> ,</ span > < span class ="n "> M</ span > < span class ="o "> =</ span > < span class ="n "> M</ span > < span class ="p "> )</ span > </ div >
536
+ < span class ="k "> return</ span > < span class ="n "> _svd_lowrank</ span > < span class ="p "> (</ span > < span class ="n "> A</ span > < span class ="p "> ,</ span > < span class ="n "> q</ span > < span class ="o "> =</ span > < span class ="n "> q</ span > < span class ="p "> ,</ span > < span class ="n "> niter</ span > < span class ="o "> =</ span > < span class ="n "> niter</ span > < span class ="p "> ,</ span > < span class ="n "> M</ span > < span class ="o "> =</ span > < span class ="n "> M</ span > < span class ="p "> )</ span >
537
537
538
538
539
539
< span class ="k "> def</ span > < span class ="nf "> _svd_lowrank</ span > < span class ="p "> (</ span > < span class ="n "> A</ span > < span class ="p "> :</ span > < span class ="n "> Tensor</ span > < span class ="p "> ,</ span > < span class ="n "> q</ span > < span class ="p "> :</ span > < span class ="n "> Optional</ span > < span class ="p "> [</ span > < span class ="nb "> int</ span > < span class ="p "> ]</ span > < span class ="o "> =</ span > < span class ="mi "> 6</ span > < span class ="p "> ,</ span > < span class ="n "> niter</ span > < span class ="p "> :</ span > < span class ="n "> Optional</ span > < span class ="p "> [</ span > < span class ="nb "> int</ span > < span class ="p "> ]</ span > < span class ="o "> =</ span > < span class ="mi "> 2</ span > < span class ="p "> ,</ span >
@@ -583,7 +583,7 @@ <h1>Source code for torch._lowrank</h1><div class="highlight"><pre>
583
583
< span class ="k "> return</ span > < span class ="n "> U</ span > < span class ="p "> ,</ span > < span class ="n "> S</ span > < span class ="p "> ,</ span > < span class ="n "> V</ span >
584
584
585
585
586
- < span class ="k "> def</ span > < span class ="nf "> pca_lowrank</ span > < span class ="p "> (</ span > < span class ="n "> A</ span > < span class ="p "> :</ span > < span class ="n "> Tensor</ span > < span class ="p "> ,</ span > < span class ="n "> q</ span > < span class ="p "> :</ span > < span class ="n "> Optional</ span > < span class ="p "> [</ span > < span class ="nb "> int</ span > < span class ="p "> ]</ span > < span class ="o "> =</ span > < span class ="kc "> None</ span > < span class ="p "> ,</ span > < span class ="n "> center</ span > < span class ="p "> :</ span > < span class ="nb "> bool</ span > < span class ="o "> =</ span > < span class ="kc "> True</ span > < span class ="p "> ,</ span >
586
+ < div class =" viewcode-block " id =" pca_lowrank " > < a class =" viewcode-back " href =" ../../generated/torch.pca_lowrank.html#torch.pca_lowrank " > [docs] </ a > < span class ="k "> def</ span > < span class ="nf "> pca_lowrank</ span > < span class ="p "> (</ span > < span class ="n "> A</ span > < span class ="p "> :</ span > < span class ="n "> Tensor</ span > < span class ="p "> ,</ span > < span class ="n "> q</ span > < span class ="p "> :</ span > < span class ="n "> Optional</ span > < span class ="p "> [</ span > < span class ="nb "> int</ span > < span class ="p "> ]</ span > < span class ="o "> =</ span > < span class ="kc "> None</ span > < span class ="p "> ,</ span > < span class ="n "> center</ span > < span class ="p "> :</ span > < span class ="nb "> bool</ span > < span class ="o "> =</ span > < span class ="kc "> True</ span > < span class ="p "> ,</ span >
587
587
< span class ="n "> niter</ span > < span class ="p "> :</ span > < span class ="nb "> int</ span > < span class ="o "> =</ span > < span class ="mi "> 2</ span > < span class ="p "> )</ span > < span class ="o "> -></ span > < span class ="n "> Tuple</ span > < span class ="p "> [</ span > < span class ="n "> Tensor</ span > < span class ="p "> ,</ span > < span class ="n "> Tensor</ span > < span class ="p "> ,</ span > < span class ="n "> Tensor</ span > < span class ="p "> ]:</ span >
588
588
< span class ="sa "> r</ span > < span class ="sd "> """Performs linear Principal Component Analysis (PCA) on a low-rank</ span >
589
589
< span class ="sd "> matrix, batches of such matrices, or sparse matrix.</ span >
@@ -684,7 +684,7 @@ <h1>Source code for torch._lowrank</h1><div class="highlight"><pre>
684
684
< span class ="k "> return</ span > < span class ="n "> _svd_lowrank</ span > < span class ="p "> (</ span > < span class ="n "> A</ span > < span class ="p "> ,</ span > < span class ="n "> q</ span > < span class ="p "> ,</ span > < span class ="n "> niter</ span > < span class ="o "> =</ span > < span class ="n "> niter</ span > < span class ="p "> ,</ span > < span class ="n "> M</ span > < span class ="o "> =</ span > < span class ="n "> M</ span > < span class ="p "> )</ span >
685
685
< span class ="k "> else</ span > < span class ="p "> :</ span >
686
686
< span class ="n "> C</ span > < span class ="o "> =</ span > < span class ="n "> A</ span > < span class ="o "> .</ span > < span class ="n "> mean</ span > < span class ="p "> (</ span > < span class ="n "> dim</ span > < span class ="o "> =</ span > < span class ="p "> (</ span > < span class ="o "> -</ span > < span class ="mi "> 2</ span > < span class ="p "> ,),</ span > < span class ="n "> keepdim</ span > < span class ="o "> =</ span > < span class ="kc "> True</ span > < span class ="p "> )</ span >
687
- < span class ="k "> return</ span > < span class ="n "> _svd_lowrank</ span > < span class ="p "> (</ span > < span class ="n "> A</ span > < span class ="o "> -</ span > < span class ="n "> C</ span > < span class ="p "> ,</ span > < span class ="n "> q</ span > < span class ="p "> ,</ span > < span class ="n "> niter</ span > < span class ="o "> =</ span > < span class ="n "> niter</ span > < span class ="p "> ,</ span > < span class ="n "> M</ span > < span class ="o "> =</ span > < span class ="kc "> None</ span > < span class ="p "> )</ span >
687
+ < span class ="k "> return</ span > < span class ="n "> _svd_lowrank</ span > < span class ="p "> (</ span > < span class ="n "> A</ span > < span class ="o "> -</ span > < span class ="n "> C</ span > < span class ="p "> ,</ span > < span class ="n "> q</ span > < span class ="p "> ,</ span > < span class ="n "> niter</ span > < span class ="o "> =</ span > < span class ="n "> niter</ span > < span class ="p "> ,</ span > < span class ="n "> M</ span > < span class ="o "> =</ span > < span class ="kc "> None</ span > < span class ="p "> )</ span > </ div >
688
688
</ pre > </ div >
689
689
690
690
</ article >
0 commit comments