线性方程组求解

求线程方程组的解:

有如下三元一次方程，求解
x + y + z = 10
x - 2y + 2z = 7
2x + y - z = -1

1. 通过矩阵求逆来解

import torch
from scipy.optimize import fsolve

# 通过矩阵求逆来解
def solve_by_inverse():
    # 按方程变量前面的系数构造方阵(3x3)
    tmat = torch.Tensor([[1, 1,  1],
                         [1, -2, 2],
                         [2, 1, -1]])
    # 对方阵求逆
    tmatInv = tmat.inverse()
    # 结果张量(3x1)
    tmatRlt = torch.Tensor([[10], [7], [-1]])
    # 逆矩阵 * 结果矩阵 = 得到方程的解
    tRlt = tmatInv.mm(tmatRlt)
    print("[x,y,z]=", tRlt)

结果：
[x, y, z]= tensor([[1.0000],
[3.0000],
[6.0000]])

2 使用scipy库的fsolve函数来求解(近似解)

def solve_by_fsolve():
    # 定义三元一次方线性方程组
    def solve_function(unsolved_value):
        x, y, z = unsolved_value[0], unsolved_value[1], unsolved_value[2]
        return [
            x + y + z - 10,     # 这里要写成 x + y + z - 10 = 0 的形式
            x - 2*y + 2*z - 7,
            2*x + y - z + 1
        ]
    # 解方程组
    solved = fsolve(solve_function, [0, 0, 0])
    print("[x, y, z]=", solved)

结果：
[x, y, z]= [1. 3. 6.]