tensorflow求导和梯度计算

最新推荐文章于 2021-12-20 00:37:51 发布

大雄没有叮当猫

最新推荐文章于 2021-12-20 00:37:51 发布

阅读量5.2k

点赞数 1

分类专栏：深度学习

本文链接：https://blog.csdn.net/u013230189/article/details/81739520

版权

1.函数求一阶导

import tensorflow as tf
tf.enable_eager_execution()
tfe=tf.contrib.eager
from math import  pi
def f(x):
    return tf.square(tf.sin(x))
assert f(pi/2).numpy()==1.0
sess=tf.Session()
grad_f=tfe.gradients_function(f)
print(grad_f(np.zeros(1))[0].numpy())

2.高阶函数求导

import numpy as np
def  f(x):
    return tf.square(tf.sin(x))

def grad(f):
    return lambda x:tfe.gradients_function(f)(x)[0]

x=tf.lin_space(-2*pi,2*pi,100)
# print(grad(f)(x).numpy())
x=x.numpy()
import matplotlib.pyplot as plt
plt.plot(x,f(x).numpy(),label="f")
plt.plot(x,grad(f)(x).numpy(),label="first derivative")#一阶导
plt.plot(x,grad(grad(f))(x).numpy(),label="second derivative")#二阶导
plt.plot(x,grad(grad(grad(f)))(x).numpy(),label="third derivative")#三阶导
plt.legend()
plt.show()