PyTorch设立计算图并自动计算(梯度下降和线性回归的实现)

numpy和pytorch实现梯度下降法 设定初始值 求取梯度 在梯度方向上进行参数的更新 使用Python实现 x = 1 learning_rate = 0.1 epochs = 100 y = lambda x : x ** 2 + 2 * x + 1 for epoch in range(epochs): dx = 2 * x + 2 x = x - learning_r
相关文章
相关标签/搜索