PyTorch torch.pinverse 関数
torch.pinversePyTorch において行列の Moore-Penrose 疑似逆行列を計算するための関数です。これはtorch.linalg.pinvのエイリアスです。
関数定義
torch.pinverse(input, rcond=1e-15)
パラメータ:
input(Tensor): 入力行列。rcond(float, オプション): 特異値の切り捨て閾値。デフォルトは 1e-15 です。
戻り値:
torch.Tensor: 行列の疑似逆行列を返します。
使用例
例
import torch
# 非正方行列を作成
A = torch.tensor([[1.0, 2.0, 3.0],
[4.0, 5.0, 6.0]], dtype=torch.float64)
# 疑似逆行列を計算
A_pinv = torch.pinverse(A)
print("行列 A:")
print(A)
print("n疑似逆行列 A^+:")
print(A_pinv)
# 非正方行列を作成
A = torch.tensor([[1.0, 2.0, 3.0],
[4.0, 5.0, 6.0]], dtype=torch.float64)
# 疑似逆行列を計算
A_pinv = torch.pinverse(A)
print("行列 A:")
print(A)
print("n疑似逆行列 A^+:")
print(A_pinv)
出力結果は次のとおりです：
矩阵 A:
tensor([[1., 2., 3.],
[4., 5., 6.]], dtype=torch.float64)
伪逆 A^+:
tensor([[-0.9444, 0.4444],
[-0.2778, 0.2778],
[ 0.3889, 0.1111]], dtype=float64)
その他の拡張