PyTorch torch.linalg.solve 関数
torch.linalg.solvePyTorch の線形代数モジュールで線形方程式を解くための関数です。AX = B を解きます。
関数定義
torch.linalg.solve(A, B, left=True, out=None)
パラメータ:
A(Tensor): 係数行列。B(Tensor): 右辺の行列またはベクトル。left(bool, オプション): True の場合、AX = B を解きます。False の場合、XA = B を解きます。デフォルトは True です。out(Tensor, オプション): 出力テンソル。
戻り値:
torch.Tensor: 連立一次方程式の解を返します。
使用例
例
import torch
# 係数行列と右辺ベクトルを作成
A = torch.tensor([[1.0, 2.0], [3.0, 4.0]])
B = torch.tensor([5.0, 11.0])
# AX = B を解く
X = torch.linalg.solve(A, B)
print("係数行列 A:")
print(A)
print("n右辺ベクトル B:")
print(B)
print("n解 X:")
print(X)
print("n検証: A @ X =")
print(A @ X)
# 係数行列と右辺ベクトルを作成
A = torch.tensor([[1.0, 2.0], [3.0, 4.0]])
B = torch.tensor([5.0, 11.0])
# AX = B を解く
X = torch.linalg.solve(A, B)
print("係数行列 A:")
print(A)
print("n右辺ベクトル B:")
print(B)
print("n解 X:")
print(X)
print("n検証: A @ X =")
print(A @ X)
出力結果は次のとおりです：
系数矩阵 A:
tensor([[1., 2.],
[3., 4.]])
右侧向量 B:
tensor([ 5., 11.])
解 X:
tensor([1., 2.])
验证: A @ X =
tensor([ 5., 11.])
その他の拡張