scipy.optimize.

rosen_hess#

scipy.optimize.rosen_hess(x)[source]#

Rosenbrock 函数的海森矩阵。

参数:
xarray_like

要计算海森矩阵的点的一维数组。

返回:
rosen_hessndarray

Rosenbrock 函数在 x 处的海森矩阵。

另请参阅

rosen, rosen_der, rosen_hess_prod

备注

rosen_hess 除了 NumPy 之外，还对兼容 Python 数组 API 标准的后端提供了实验性支持。请考虑通过设置环境变量 SCIPY_ARRAY_API=1 并提供 CuPy、PyTorch、JAX 或 Dask 数组作为数组参数来测试这些功能。以下后端和设备（或其他功能）组合受支持。

库

CPU

GPU

NumPy

✅

不适用

CuPy

不适用

✅

PyTorch

✅

✅

JAX

⛔

⛔

Dask

✅

不适用

有关更多信息，请参阅 对数组 API 标准的支持。

示例

>>> import numpy as np
>>> from scipy.optimize import rosen_hess
>>> X = 0.1 * np.arange(4)
>>> rosen_hess(X)
array([[-38.,   0.,   0.,   0.],
       [  0., 134., -40.,   0.],
       [  0., -40., 130., -80.],
       [  0.,   0., -80., 200.]])
在本页