这个算法虽然算是新思想,但是代码构造复杂,并且精度不高,所以只放在上面作为参考。这个算法最难的地方在于要设计一个单输入,多输出的MLP,同时还要求出Jacobi矩阵,目前还没有针对单输入,多输出的函数封装好的自动求梯度的函数,所以求解时间长。
import numpy as np
import matplotlib.pyplot as plt
import torch
import time
import torch.nn as
这个算法虽然算是新思想,但是代码构造复杂,并且精度不高,所以只放在上面作为参考。这个算法最难的地方在于要设计一个单输入,多输出的MLP,同时还要求出Jacobi矩阵,目前还没有针对单输入,多输出的函数封装好的自动求梯度的函数,所以求解时间长。
import numpy as np
import matplotlib.pyplot as plt
import torch
import time
import torch.nn as