恢复模型时如何在张量流中引用全连接层的输出?

时间:2018-10-07 10:51:10

标签: tensorflow tensorflow-serving

如果我要连接一个如下所示的完全连接的层,并且此输出是我在还原模型时要引用的logit,但是name中没有tf.contrib.layers.fully_connected参数可用。以下是我要执行的操作,但会引发错误

logits = tf.contrib.layers.fully_connected(outl[str(i)], self.classes, activation_fn = None, 
                                                           weights_initializer = self.initializer,
                                                          name = 'mlogits')

fully_connected() got an unexpected keyword argument 'name'

此完全连接的层的输出的张量名称是什么?

1 个答案:

答案 0 :(得分:0)

您可以使用.name属性访问输出张量的完整名称:

print(logits.name)

这是您的输出张量的完整名称