9 lines
345 B
Python
9 lines
345 B
Python
import torch
|
||
|
||
# 构建超定方程组 Ax=b (3方程2未知数)
|
||
A = torch.tensor([[2.0, 3], [1, 4], [3, 1]]) # 3x2矩阵
|
||
b = torch.tensor([5.0, 6, 4]).reshape(-1,1) # 3x1向量
|
||
|
||
# 解法1：正规方程 (A^T A)^-1 A^T b
|
||
solution = torch.linalg.lstsq(A, b).solution # PyTorch内置最小二乘
|
||
print(f"最小二乘解:\n{solution.numpy()}") |