3.4tensorflow2.x自动求导原理函数详解

自己开发了一个股票智能分析软件,功能很强大,需要的点击下面的链接获取:

https://www.cnblogs.com/bclshuai/p/11380657.html

1.1  tensorflow2.x自动求导

1.1.1         自动求导GradientTape类

GradientTape的作用就是用于自动求导,需要有自变量x和因变量y,调用gradient(y,x)就可以求导,在GradientTape定义的上下文中,会默认观察自变量x,并进行记录,需要占用内存等资源,所以可以精确指定观察变量,减少资源。自变量可以是多个,gradient(y,[x,p,g]) ,计算的结果就是多个变量的张量组合。也可以通过嵌套的方式实现高阶求导。

1.1.2         GradientTape类定义

class GradientTape(object):

(1)初始化构造函数

 def __init__(self, persistent=False, watch_accessed_variables=True):

#persistent变量,默认情况下,调用一次求导之后,GradientTape所持有的资源就会被释放,不能再执行,如果需要持续求导,persistent默认为False,,也就是g只能调用一次,如果指定persistent为true,则可以多次求导。

#watch_accessed_variables,是否要指定观察的变量,默认为True,默认观察并记录上下文中的自变量,g.watch(x)可以不写,自动观察求导中的变量。如果要对监视变量进行精细控制,可以通过将watch_accessed_variables = False传递给tape,手动指定观察变量,避免全部观察记录,减少不必要的资源消耗。

实例

x = tf.Variable(2.0)

w = tf.Variable(5.0)

with tf.GradientTape(

    watch_accessed_variables=False, persistent=True) as tape:

  tape.watch(x)

  y = x ** 2  # Gradients will be available for `x`.

  z = w ** 3  # No gradients will be available as `w` isn't being watched.

dy_dx = tape.gradient(y, x)

print(dy_dx)

 

# No gradients will be available as `w` isn't being watched.

dz_dy = tape.gradient(z, w)

print(dz_dy)

(2)进入和退出函数

输入一个上下文,在该上下文中将操作记录在此tape上

def __enter__(self):

退出记录上下文,不再跟踪其他操作

def __exit__(self, typ, value, traceback):

(3)观察变量函数

指定观察变量,默认自动观察求导自变量,watch_accessed_variables = False时才需要手动指定。

def watch(self, tensor):

(1)   停止在tape上记录操作

暂时停止在该tape上进行记录操作。此上下文管理器处于活动状态时执行的操作不会记录在tape上。这对于减少通过跟踪所有计算而使用的内存很有用。

def stop_recording(self):

实例:

import tensorflow as tf
x = tf.Variable(4.0)
with tf.GradientTape(persistent=True) as tape:
    y = tf.pow(x, 2)
    z=tape.gradient(y,x)
    #with tape.stop_recording():# The gradient computation below is not traced, saving memory.
    y+=x
    dy_dx = tape.gradient(y, x)
print(z)
print(dy_dx)

输出结果:

tf.Tensor(8.0, shape=(), dtype=float32)

tf.Tensor(9.0, shape=(), dtype=float32)

 

如果加上stop_recording

import tensorflow as tf
x = tf.Variable(4.0)
with tf.GradientTape(persistent=True) as tape:
    y = tf.pow(x, 2)
    z=tape.gradient(y,x)
    with tape.stop_recording():# The gradient computation below is not traced, saving memory.
        y+=x
        dy_dx = tape.gradient(y, x)
print(z)
print(dy_dx)

输出结果为

tf.Tensor(8.0, shape=(), dtype=float32)

None

可见stop_recording作用是停止记录变量的计算,不记录结果。求导结果为None;

(2)   重置求导记录函数

清除此tape中存储的所有信息。等效于退出并重新进入tape上下文管理器。

def reset(self):

import tensorflow as tf
x=tf.Variable(initial_value=[1.,2.,3,])
with tf.GradientTape() as t:
  loss = tf.pow(x,2)
with tf.GradientTape() as t:
  loss += x;
z=t.gradient(loss, x)  # Only differentiates other_loss_fn, not loss_fn
print(z)

# The following is equivalent to the above
with tf.GradientTape() as t:
  loss = tf.pow(x,2)#调用了reset,这里的记录会被清除
  #t.reset()注释,loss=x*x+x,不注释,会将x*x的记录清除,loss=x
  loss += x
z=t.gradient(loss, x)  # Only differentiates other_loss_fn, not loss_fn
print(z)

输出结果

tf.Tensor([1. 1. 1.], shape=(3,), dtype=float32)# 调用reset清除记录x*x

tf.Tensor([3. 5. 7.], shape=(3,), dtype=float32)# 不清除x*x

(6)返回观察的变量

该函数监视要求导的变量,可以是一个tensor[x],也可以是多个参数组成的列表[x,y],如果watch_accessed_variables定义为True,则可以不调用该函数,可以自动求导。

def watched_variables(self):

(7)求导函数

    def gradient(self,

               target,# 求导的函数y,可以理解为要求导的因变量y

               sources,# 要对哪一个自变量x求导,可以是多个组合

               output_gradients=None,求导后输出结果乘以的系数

               unconnected_gradients=UnconnectedGradients.NONE):# 它是一个可选参数,有两个值,“none”和“zero”,none是它的默认值,表示当我们的target(因变量y)与sources(自变量x)之间没有关系时,返回NONE

(8)雅克比矩阵

在向量微积分中,雅可比矩阵是一阶偏导数以一定方式排列成的矩阵。

    def jacobian(self,

               target, #求导的函数y,可以理解为要求导的因变量y

               sources,自变量x,维度大于等于2的张量

               unconnected_gradients=UnconnectedGradients.NONE,

               parallel_iterations=None,# 控制并行调度的迭代次数,可以用来控制总的内存使用量

               experimental_use_pfor=True):

实例

with tf.GradientTape() as g:

  x = tf.constant([[1., 2.], [3., 4.]], dtype=tf.float32)

  g.watch(x)

  y = x * x#一阶导数是2*x

batch_jacobian = g.batch_jacobian(y, x)

# batch_jacobian is [[[2,  0], [0,  4]], [[6,  0], [0,  8]]]

1.1.3         自动求导的步骤

(1)创建一个GradientTape对象:g=tf.GradientTape()

(2)监视watch要求导的变量:g.watch(x)

(3)对函数进行求导:g.gredient(y,x)

1.1.4         自动求导实例

(1)高阶导数嵌套实例

先初始化一个变量,然后定义变量的函数,然后用tf.GradientTape()求一阶导数和二阶导数。

import tensorflow as tf
#定义并初始化变量
x=tf.Variable(initial_value=[[1.,2.,3.],[4.,5.,6.]])
#创建GradientTape对象
with tf.GradientTape() as g1:
    #指定观察变量
   # g1.watch(x)
    with tf.GradientTape(persistent=True) as g2:#嵌套高阶求导
       # g2.watch(x)
        y=x*x
        z=tf.sqrt(y+1)
    y1=g2.gradient(y,x)#一阶导数
    z1x=g2.gradient(z,x)#z对于x的导数,因为g2的persistent参数设置为true,所以可以调用两次,否则报错。
y2=g1.gradient(y1,x)#二阶导数

print(y)
print(y1)
print(y2)
print(z1x)

 

输出结果:

tf.Tensor(

[[ 1.  4.  9.]

 [16. 25. 36.]], shape=(2, 3), dtype=float32)

tf.Tensor(

[[ 2.  4.  6.]

 [ 8. 10. 12.]], shape=(2, 3), dtype=float32)

tf.Tensor(

[[2. 2. 2.]

 [2. 2. 2.]], shape=(2, 3), dtype=float32)

tf.Tensor(

[[0.70710677 0.8944272  0.94868326]

 [0.97014254 0.9805807  0.9863939 ]], shape=(2, 3), dtype=float32)

(2)多变量和输出系数求导实例

import tensorflow as tf

x = tf.Variable(initial_value=[1.0, 2.0, 3.0])
y = tf.Variable(initial_value=[2.0, 4.0, 6.0])

with tf.GradientTape(persistent=True) as g:
    g.watch(x)
    g.watch(y)
    z = tf.pow(x, 2) + tf.pow(y, 2)

output_gradients = tf.Variable([0.2, 0.6, 0.2])#输出结果相乘的系数
dz = g.gradient(z, [x, y], output_gradients=output_gradients)  # 给每一个元素施加不同的权重
print(dz)
'''运行结果为:
[<tf.Tensor: id=64, shape=(3,), dtype=float32, numpy=array([0.4, 2.4, 1.2],
  dtype=float32)>,
 <tf.Tensor: id=40, shape=(3,), dtype=float32, numpy=array([0.8, 4.8, 2.4],
  dtype=float32)>]
'''
'''
本来的结果是dz_dx=[2,4,6], 分别乘以权重[0.2,0.6,0.2]之后,得到[0.4,2.4,1.2]
本来的结果是dz_dy=[4,8,12],分别乘以权重[0.2,0.6,0.2]之后,得到[0.8,4.8,2.4]
'''

 

1.1.5         参考文献

https://dengbocong.blog.csdn.net/article/details/108044938

https://blog.csdn.net/qq_27825451/article/details/89556703

https://tensorflow.google.cn/api_docs/python/tf/GradientTape#reset

posted @ 2021-01-06 21:24  一字千金  阅读(890)  评论(0编辑  收藏  举报