归档: 2021

0

PyTorch-SGD-求函数最优值

面试遇到的一个基础问题:使用PyTorch求一个函数的最优值代码如下: import torch.optim as optim import torch # 参数为x,(1,1) x = torch.randn([1],requires_grad=True) def function(x): return x**2 - 2*x + 1 optimizer = optim.SGD([x