tensorflow一些参数的详细讲解分享(tensorflow1的哦)
tf1:
tf.GradientTape()函数:
tf提供的自动求导函数
x = tf.Variable(initial_value=3.)
with tf.GradientTape() as tape: # 在 tf.GradientTape() 的上下文内,所有计算步骤都会被记录以用于求导
y = tf.square(x)
y_grad = tape.gradient(y, x) # 计算y关于x的导数
tf.GradientTape() 是一个自动求导的记...
原创
2021-12-19 18:20:49 ·
2084 阅读 ·
0 评论