jax.scipy.optimize.minimize

目录

jax.scipy.optimize.minimize#

jax.scipy.optimize.minimize(fun, x0, args=(), *, method, tol=None, options=None)[源代码][源代码]#

最小化一个或多个变量的标量函数。

此函数的API与SciPy匹配，但有一些小的偏差：

  • 当需要时，fun 的梯度会自动使用 JAX 的自动微分支持进行计算。

  • method 参数是必需的。你必须指定一个求解器。

  • SciPy 接口中的各种可选参数尚未实现。

  • 由于线搜索实现的不同，优化结果可能与SciPy有所不同。

minimize 支持 jit() 编译。它目前还不支持微分或以多维数组形式的参数，但这两者的支持都在计划中。

参数:

  • fun (Callable) – 要最小化的目标函数，fun(x, *args) -> float，其中 x 是一个形状为 (n,) 的一维数组，args 是一个包含完全指定函数所需固定参数的元组。fun 必须支持微分。

  • x0 (jax.Array) – 初始猜测。大小为 (n,) 的实数元素数组，其中 n 是独立变量的数量。

  • args (tuple) – 传递给目标函数的额外参数。

  • method (str) – 求解器类型。目前仅支持 "BFGS"。

  • tol (float | None) – 终止容差。如需详细控制，请使用特定求解器选项。

  • options (Mapping[str, Any] | None) – 求解器选项的字典。所有方法都接受以下通用选项： - maxiter (int): 要执行的最大迭代次数。根据方法的不同，每次迭代可能使用多次函数评估。

返回:

一个 OptimizeResults 对象。

返回类型:

OptimizeResults

目录