首页 > 解决方案 > 哪些类型的操作将/不会插入 Tensorflow 2 中的计算图?

问题描述

基于这篇文章,我们可以改变tf.variable

threshold = 3
a = np.array([1,2,3,4,5,6])
b = tf.Variable(a)
b = tf.where(b >= threshold, 199, b)

b=tf.where(b>=3, 199, b)插入计算图并影响b反向传播的梯度吗?

或更一般的问题:哪些类型的操作将/不会将计算图插入Tensorflow 2

标签: python-3.xtensorflow2.0gradientautodiff

解决方案


推荐阅读