import tensorflow as tf
x = tf.constant(5.)
def cal_grad(x):
with tf.GradientTape() as tape:
tape.watch(x)
y = x**2 - 2.*x + 5.
grad = tape.gradient(y, x)
return grad
for i in range(20):
x = x - 0.1*cal_grad(x)
print(x)
梯度下降的通俗理解
最新推荐文章于 2022-09-27 21:15:38 发布