奋斗飞鸟

文章
4
资源
0
加入时间
2年10月21天

pytorch利用梯度下降求方程的最小值_day2

pytorch我们需要求解上面函数f(x,y)的最小值,方法如下。import numpy as npfrom mpl_toolkits.mplot3d import Axes3Dfrom matplotlib import pyplot as pltimport torchdef himmelblau(x): # 利用梯度下降计算这个函数的最小值 return (x[0] ** 2 + x[1] - 11) ** 2 + (x[0] + x[1] ** 2 - 7) ** 2