当前位置:   article > 正文

pytorch利用梯度下降求方程的最小值_day2_pytorch 梯度下降法 最小化

pytorch 梯度下降法 最小化

pytorch

在这里插入图片描述
我们需要求解上面函数f(x,y)的最小值,方法如下。

import numpy as np
from mpl_toolkits.mplot3d import Axes3D
from matplotlib import pyplot as plt
import torch


def himmelblau(x):  # 利用梯度下降计算这个函数的最小值
    return (x[0] ** 2 + x[1] - 11) ** 2 + (x[0] + x[
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/Gausst松鼠会/article/detail/135795
推荐阅读
相关标签
  

闽ICP备14008679号