TensorFlow2.0 GradientTape(2)

参数 persistent是设置持久Tape内容。正常的情况下,此值为False。当调用tape.gradient之后就会释放Tape中的内容。也就是说,如果要对变量x求两个函数的导数,是报错的。
但是如果设置此值为True,就可以连续的计算导数。比如下面的例子。不过此种形式下,需要显式的删除Tape。

import tensorflow as tf
import numpy as np
import matplotlib.pyplot as plt

x=tf.linspace(0.0,3.0,150)
with tf.GradientTape(persistent=True) as tape:
    tape.watch(x)
    y=x*x
    z=tf.math.log(x)
  
grads = tape.gradient(y,x)
grads_z=tape.gradient(z,x)
del tape
plt.plot(x,y,color='r')
plt.plot(x,z,color='b')
plt.plot(x,grads,color='g')
plt.plot(x,grads_z,color='y')

plt.show()
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值