PyTorch torch.triangular_solve 関数
Pytorch torch リファレンスマニュアルtorch.triangular_solveは PyTorch で三角連立一次方程式を解くための関数です。この関数は AX = B を解きます。ここで A は三角行列です。
関数定義
torch.triangular_solve(A, B, upper, transpose, unitriangular)
パラメータの説明
A: 係数行列（正方行列である必要があります）B: 右側の行列またはベクトルupper: A が上三角行列かどうか（デフォルトは True）transpose: A を転置するかどうか（デフォルトは False）unitriangular: 単位三角行列を使用するかどうか（デフォルトは False）
使用例
実例
import torch
# 上三角係数行列を作成
A = torch.tensor([[3.0, 1.0, 2.0],
[0.0, 2.0, 1.0],
[0.0, 0.0, 1.0]])
# 右側のベクトル
B = torch.tensor([9.0, 5.0, 2.0])
# AX = B を解く
X = torch.triangular_solve(B.unsqueeze(1), A, upper=True)
print("解 X:")
print(X.solution)
# 上三角係数行列を作成
A = torch.tensor([[3.0, 1.0, 2.0],
[0.0, 2.0, 1.0],
[0.0, 0.0, 1.0]])
# 右側のベクトル
B = torch.tensor([9.0, 5.0, 2.0])
# AX = B を解く
X = torch.triangular_solve(B.unsqueeze(1), A, upper=True)
print("解 X:")
print(X.solution)
出力結果は次のとおりです：
解 X:
tensor([[1.],
[2.],
[2.]])
その他の拡張