PyTorch torch.linalg.inv 関数
Pytorch torch リファレンスマニュアルtorch.linalg.invは PyTorch 線形代数モジュールで行列の逆行列を計算するための関数です。これはtorch.inverseの推奨代替関数です。
関数定義
torch.linalg.inv(A)
パラメータ:
A(Tensor): 入力正方行列で、可逆である必要があります。
戻り値:
torch.Tensor: 行列の逆行列を返します。
使用例
例
import torch
# 可逆行列を作成
A = torch.tensor([[1.0, 2.0],
[3.0, 4.0]])
# 逆行列を計算
A_inv = torch.linalg.inv(A)
print("行列 A:")
print(A)
print("n逆行列 A^-1:")
print(A_inv)
print("n検証: A @ A^-1 =")
print(A @ A_inv)
# 可逆行列を作成
A = torch.tensor([[1.0, 2.0],
[3.0, 4.0]])
# 逆行列を計算
A_inv = torch.linalg.inv(A)
print("行列 A:")
print(A)
print("n逆行列 A^-1:")
print(A_inv)
print("n検証: A @ A^-1 =")
print(A @ A_inv)
出力結果は次のとおりです。
矩阵 A:
tensor([[1., 2.],
[3., 4.]])
逆矩阵 A^-1:
tensor([[-2.0000, 1.0000],
[ 1.5000, -0.5000]])
验证: A @ A^-1 =
tensor([[1., 0.],
[0., 1.]])
その他の拡張