python - 我可以在不应用输入的情况下获得张量相对于输入的梯度吗？

Question

例如，我需要计算cross_entropy相对于的梯度x，但我需要对梯度函数应用另一个值。

那是：

f'(x)|x = x_t

我认为tf.gradients()函数只会在x = x. 那么 tensorflow 是否提供了这些功能？

Answer 1

的结果tf.gradients是张量（一般的张量列表），而不是浮点值。在某种程度上，这个张量是一个函数：它可以在任何点进行评估。客户端只需要提供所需的输入值。

例子：

features = 3
n_samples = 10
hidden = 1

X = tf.placeholder(dtype=tf.float32, shape=[n_samples, features])
Y = tf.placeholder(dtype=tf.float32, shape=[n_samples])

W = tf.Variable(np.ones([features, hidden]), dtype=tf.float32, name="weight")
b = tf.Variable(np.ones([hidden]), dtype=tf.float32, name="bias")

pred = tf.add(tf.matmul(X, W), b)
cost = tf.reduce_mean(tf.pow(pred - Y, 2))

dc_dw, dc_db = tf.gradients(cost, [W, b])

session.run(tf.global_variables_initializer())

# Let's compute `dc_dw` at `ones` matrix.
print(dc_dw.eval(feed_dict={X: np.ones([n_samples, features]), 
                            Y: np.ones([n_samples])}))