Tensorflow - 无法将操作转换为Tensor

时间:2017-12-11 09:59:03

标签: tensorflow

我想计算一个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)))

我该如何解决这个问题?

1 个答案:

答案 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')