Tensorflow学习笔记(1) 利用函数修饰符@tf.custom_gradients自定义函数梯度
利用函数修饰符@tf.custom_gradients自定义函数梯度在tensorflow v1.12中,新定义了一个修饰符函数tf.custom_gradients,用于封装自定义的函数-导数对。有时候我们想使用tensorflow去计算一些函数的梯度,但会碰到如下情况def log1pexp(x): e = tf.exp(x) return tf.log(1+e)x ...
原创
2019-01-09 22:10:01 ·
3966 阅读 ·
3 评论