猿问

tensorflow.nn.conv1d 有梯度吗?

我试图用get_gradient_function()的tensorflow.nn.conv1d是这样的:


import tensorflow as tf

from tensorflow.python.framework.ops import get_gradient_function


d = tf.constant([1, 0, 2, 3, 0, 1, 1], dtype=tf.float32, name='d')

k = tf.constant([2, 1, 3], dtype=tf.float32, name='k')


data = tf.reshape(d, [1, int(d.shape[0]), 1], name='data')

kernel = tf.reshape(k, [int(k.shape[0]), 1, 1], name='kernel')


conv = tf.nn.conv1d(data, kernel, 1, 'SAME', name='conv')


with tf.Session() as sess:

    print (sess.run(conv))


op = tf.get_default_graph().get_operation_by_name('conv')

print(get_gradient_function(op))

我在倒数第二行收到以下错误。


KeyError:“名称‘conv’指的是不在图中的操作。”


慕丝7291255
浏览 197回答 1
1回答
随时随地看视频慕课网APP

相关分类

Python
我要回答