一睹为快,自动梯度求导

一睹为快,自动梯度求导

笔者不想过多的谈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)
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容