本文简要介绍python语言中 torch.linalg.pinv
的用法。
用法:
torch.linalg.pinv(A, rcond=1e-15, hermitian=False, *, out=None) → Tensor
A(Tensor) -形状为
(*, m, n)
的张量,其中*
是零个或多个批次维度。rcond(float或者Tensor,可选的) -确定奇异值何时为零的容差值如果它是
torch.Tensor
,则其形状必须可广播为torch.svd()
返回的A
的奇异值的形状。默认值:1e-15
。hermitian(bool,可选的) -指示
A
如果是复数是 Hermitian,如果是实数是对称的。默认值:False
。
out(Tensor,可选的) -输出张量。如果
None
则忽略。默认值:None
。计算矩阵的伪逆(Moore-Penrose逆)。
伪逆可能是defined algebraically,但理解它在计算上更方便through the SVD
支持 float、double、cfloat 和 cdouble dtypes 的输入。还支持批量矩阵,如果
A
是批量矩阵,则输出具有相同的批量维度。如果
hermitian
= True
,A
假设是 Hermitian 如果复数或对称如果实数,但内部不检查。相反,在计算中仅使用矩阵的下三角部分。低于指定
rcond
阈值的奇异值(或当hermitian
= True
时的特征值的范数)被视为零并在计算中被丢弃。注意
此函数使用
torch.linalg.svd()
ifhermitian
= False
和torch.linalg.eigh()
ifhermitian
= True
。对于 CUDA 输入,此函数将该设备与 CPU 同步。注意
如果可能,考虑使用
torch.linalg.lstsq()
将左侧矩阵乘以伪逆矩阵,如下所示:torch.linalg.lstsq(A, B).solution == A.pinv() @ B
在可能的情况下,总是首选使用
lstsq()
,因为它比显式计算伪逆更快且数值更稳定。警告
该函数内部使用
torch.linalg.svd()
(或torch.linalg.eigh()
,当hermitian
= True
时),因此其导数与这些函数具有相同的问题。有关更多详细信息,请参阅torch.linalg.svd()
和torch.linalg.eigh()
中的警告。例子:
>>> A = torch.randn(3, 5) >>> A tensor([[ 0.5495, 0.0979, -1.4092, -0.1128, 0.4132], [-1.1143, -0.3662, 0.3042, 1.6374, -0.9294], [-0.3269, -0.5745, -0.0382, -0.5922, -0.6759]]) >>> torch.linalg.pinv(A) tensor([[ 0.0600, -0.1933, -0.2090], [-0.0903, -0.0817, -0.4752], [-0.7124, -0.1631, -0.2272], [ 0.1356, 0.3933, -0.5023], [-0.0308, -0.1725, -0.5216]]) >>> A = torch.randn(2, 6, 3) >>> Apinv = torch.linalg.pinv(A) >>> torch.dist(Apinv @ A, torch.eye(3)) tensor(8.5633e-07) >>> A = torch.randn(3, 3, dtype=torch.complex64) >>> A = A + A.T.conj() # creates a Hermitian matrix >>> Apinv = torch.linalg.pinv(A, hermitian=True) >>> torch.dist(Apinv @ A, torch.eye(3)) tensor(1.0830e-06)
参数:
关键字参数:
相关用法
- Python PyTorch pixel_shuffle用法及代码示例
- Python PyTorch pixel_unshuffle用法及代码示例
- Python PyTorch positive用法及代码示例
- Python PyTorch promote_types用法及代码示例
- Python PyTorch powerSGD_hook用法及代码示例
- Python PyTorch pca_lowrank用法及代码示例
- Python PyTorch profile用法及代码示例
- Python PyTorch put_metric用法及代码示例
- Python PyTorch pad_sequence用法及代码示例
- Python PyTorch pow用法及代码示例
- Python PyTorch phase_vocoder用法及代码示例
- Python PyTorch pop用法及代码示例
- Python PyTorch prepare用法及代码示例
- Python PyTorch polar用法及代码示例
- Python PyTorch poisson用法及代码示例
- Python PyTorch pack_sequence用法及代码示例
- Python PyTorch pad用法及代码示例
- Python PyTorch pad_packed_sequence用法及代码示例
- Python PyTorch polygamma用法及代码示例
- Python PyTorch permute用法及代码示例
- Python PyTorch prod用法及代码示例
- Python PyTorch prof用法及代码示例
- Python PyTorch frexp用法及代码示例
- Python PyTorch jvp用法及代码示例
- Python PyTorch cholesky用法及代码示例
注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.linalg.pinv。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。