温馨提示×

温馨提示×

您好,登录后才能下订单哦!

密码登录×
登录注册×
其他方式登录
点击 登录注册 即表示同意《亿速云用户服务条款》

tensorflow 2中自动求导的机制是怎样的

发布时间:2021-12-23 16:29:08 来源:亿速云 阅读:126 作者:柒染 栏目:大数据

这篇文章将为大家详细讲解有关tensorflow 2中自动求导的机制是怎样的,文章内容质量较高,因此小编分享给大家做个参考,希望大家阅读完这篇文章后对相关知识有一定的了解。

import tensorflow as tf
import numpy as np 
 
# f(x) = a*x**2 + b*x + c的导数
x = tf.Variable(0.0, name='x', dtype=tf.float32)
a = tf.constant(1.0)
b = tf.constant(-1.5)
c = tf.constant(1.0)


with tf.GradientTape() as tape:
    y = a * tf.pow(x,2) + b * x + c

dy_dx = tape.gradient(y,x)
print(dy_dx)

可以计算出来dy_dx = 2*a*x +b = 2*1.0*0.0 + (-1.5) = -1.5

我们需要了解的是,

(1) GradienTape如何跟踪到 y = a * tf.pow(x,2) + b *x + c中间的函数变换路径。

(2)基本函数块的导数是如何求出来。

(3)复合函数求导的规则是如何实现出来。

关于tensorflow 2中自动求导的机制是怎样的就分享到这里了,希望以上内容可以对大家有一定的帮助,可以学到更多知识。如果觉得文章不错,可以把它分享出去让更多的人看到。

向AI问一下细节

免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。

AI