我想计算一个Operation和Tensor输出之间的成对欧几里德距离。我使用了建议here的代码。这是我的代码的要点:
# Suppose logits has shape [32, 128]
logits = tf.get_default_graph().get_operation_by_name('Tanh')
y = tf.placeholder(tf.float32, shape=[10, 128])
m1, m2, k = 32, tf.shape(y)[0], latent_dim
# Get the pairwise distances
p1 = tf.matmul(tf.expand_dims(tf.reduce_sum(tf.square(logits), 1), 1),
tf.ones(shape=(1, m2)))
p2 = tf.transpose(tf.matmul(
tf.reshape(tf.reduce_sum(tf.square(y), 1), shape=[-1, 1]),
tf.ones(shape=(m1, 1)),
transpose_b=True
))
distance_predictions = tf.sqrt(tf.add(p1, p2) - 2 *
tf.matmul(logits, y, transpose_b=True))
但是我收到以下错误:
TypeError: Can't convert Operation '.../Tanh' to Tensor (target dtype=None, name=u'x', as_ref=False)
对于这一行:
p1 = tf.matmul(tf.expand_dims(tf.reduce_sum(tf.square(logits), 1), 1),
tf.ones(shape=(1, m2)))
我该如何解决这个问题?
答案 0 :(得分:4)
通过调用tf.get_default_graph().get_operation_by_name
,我得到了计算tanh激活的操作。但我需要的是该操作的输出,我可以通过调用tf.get_default_graph().get_tensor_by_name
找到它。
因此修复是用
替换第一行 logits = tf.get_default_graph().get_tensor_by_name('Tanh:0')