一睹为快,自动梯度求导

一睹为快,自动梯度求导

笔者不想过多的谈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)

你可能感兴趣的:(一睹为快,自动梯度求导)