使用python,pytorch求海森Hessian矩陣

  考慮一個函數$y=f(\textbf{x}) (R^n\rightarrow R)$,y的Hessian矩陣定義如下:   考慮一個函數:$$f(x)=b^Tx+\frac{1}{2}x^{T}Ax\\其中b^T=[1,3,5], A在代碼中可讀,可以自定義$$    求該函數在x = [0,0,0]處海森矩陣值的python代碼如下:   本代碼需要用到torch.autograd包中的核心
相關文章
相關標籤/搜索