@@ -494,7 +494,7 @@ <h1>Source code for torch._lowrank</h1><div class="highlight"><pre>
494
494
< span class ="k "> return</ span > < span class ="n "> Q</ span >
495
495
496
496
497
- < 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 >
497
+ < 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 >
498
498
< 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 >
499
499
< span class ="sa "> r</ span > < span class ="sd "> """Return the singular value decomposition ``(U, S, V)`` of a matrix,</ span >
500
500
< span class ="sd "> batches of matrices, or a sparse matrix :math:`A` such that</ span >
@@ -539,7 +539,7 @@ <h1>Source code for torch._lowrank</h1><div class="highlight"><pre>
539
539
< 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 >
540
540
< 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 >
541
541
< 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 >
542
- < 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 >
542
+ < 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 >
543
543
544
544
545
545
< 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 >
@@ -589,7 +589,7 @@ <h1>Source code for torch._lowrank</h1><div class="highlight"><pre>
589
589
< 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 >
590
590
591
591
592
- < 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 >
592
+ < 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 >
593
593
< 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 >
594
594
< span class ="sa "> r</ span > < span class ="sd "> """Performs linear Principal Component Analysis (PCA) on a low-rank</ span >
595
595
< span class ="sd "> matrix, batches of such matrices, or sparse matrix.</ span >
@@ -690,7 +690,7 @@ <h1>Source code for torch._lowrank</h1><div class="highlight"><pre>
690
690
< 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 >
691
691
< span class ="k "> else</ span > < span class ="p "> :</ span >
692
692
< 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 >
693
- < 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 >
693
+ < 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 >
694
694
</ pre > </ div >
695
695
696
696
</ article >
0 commit comments