一睹为快,自动梯度求导
笔者不想过多的谈TF有什么用,只是提供一个入门TF的教程
本节目录
- 自动梯度求导
自动梯度求导
自动梯度求导可以说是所有的深度学习框架所必须的工具,无论是TF
,Torch
等。我们首先就来介绍一下TF的自动梯度求导机制:
import tensorflow as tf
x = tf.Variable(10.0,dtype = tf.float32)#声明变量x,并赋值为10,类型是float31
with tf.GradientTape() as tape: #自动求导记录
y = x**2 + x + 1 #记录y和x关系
gradient = tape.gradient(y,x) #进行自动梯度求导
print(gradient) #打印结果
运行结果如下
tf.Tensor(21.0, shape=(), dtype=float32)
网友评论