PyTorch最小化
Pytorch-minimize表示一组实用程序,用于最小化PyTorch中一个或多个变量的标量函数。 它受到SciPy的optimize模块和MATLAB的极大启发。 与SciPy和MATLAB不同,无需提供jacobian和hessian函数来进行pytorch最小化求解器,并且永远不会使用数值近似。 CPU和CUDA均受支持。 目前,仅实现了不受约束的最小化例程。
作者:鲁本·费曼(Reuben Feinman)
乍看上去:
import torch
from fmin import minimize
def rosen ( x ):
return torch . sum ( 100 * ( x [..., 1 :] - x [..., : - 1 ] ** 2 ) ** 2
+ ( 1 - x [...
1