小编Sri*_*ana的帖子

为什么Tensorflow重塑tf.reshape()会破坏渐变的流动?

我正在创建一个tf.Variable(),然后使用该变量创建一个简单的函数,然后我使用tf.reshape()展平原始变量,然后我在函数和展平变量之间使用tf.gradients().为什么返回[无].

var = tf.Variable(np.ones((5,5)), dtype = tf.float32)
f = tf.reduce_sum(tf.reduce_sum(tf.square(var)))
var_f = tf.reshape(var, [-1])
print tf.gradients(f,var_f)
Run Code Online (Sandbox Code Playgroud)

执行上面的代码块返回[None].这是一个错误吗?请帮忙!

python tensorflow

6
推荐指数
1
解决办法
2090
查看次数

标签 统计

python ×1

tensorflow ×1