@@ -486,7 +486,7 @@ <h1>Source code for torch._lowrank</h1><div class="highlight"><pre>
486
486
< span class ="k "> return</ span > < span class ="n "> Q</ span >
487
487
488
488
489
- < 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 >
489
+ < 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 >
490
490
< 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 >
491
491
< span class ="sa "> r</ span > < span class ="sd "> """Return the singular value decomposition ``(U, S, V)`` of a matrix,</ span >
492
492
< span class ="sd "> batches of matrices, or a sparse matrix :math:`A` such that</ span >
@@ -531,7 +531,7 @@ <h1>Source code for torch._lowrank</h1><div class="highlight"><pre>
531
531
< 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 >
532
532
< 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 >
533
533
< 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 >
534
- < 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 >
534
+ < 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 >
535
535
536
536
537
537
< 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 >
@@ -581,7 +581,7 @@ <h1>Source code for torch._lowrank</h1><div class="highlight"><pre>
581
581
< 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 >
582
582
583
583
584
- < 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 >
584
+ < 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 >
585
585
< 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 >
586
586
< span class ="sa "> r</ span > < span class ="sd "> """Performs linear Principal Component Analysis (PCA) on a low-rank</ span >
587
587
< span class ="sd "> matrix, batches of such matrices, or sparse matrix.</ span >
@@ -682,7 +682,7 @@ <h1>Source code for torch._lowrank</h1><div class="highlight"><pre>
682
682
< 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 >
683
683
< span class ="k "> else</ span > < span class ="p "> :</ span >
684
684
< 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 >
685
- < 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 >
685
+ < 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 >
686
686
</ pre > </ div >
687
687
688
688
</ article >
0 commit comments