如何知道从Graph中的张量名称列表中选择哪个张量?

时间:2018-07-10 13:54:03

标签: tensorflow tensorflow-lite

我正在尝试将线性分类器导出为tflite格式。这就是为什么我需要从Graph中的张量列表名称中选择一个张量。

例如,对于导出DNN分类器模型,选择了以下输入和输出张量:

input_tensor = sess.graph.get_tensor_by_name("dnn/input_from_feature_columns/input_layer/concat:0")
    input_tensor.set_shape([1, 4])
    out_tensor = sess.graph.get_tensor_by_name("dnn/logits/BiasAdd:0")
    out_tensor.set_shape([1, 3])

但是对于线性分类器,我不知道要使用哪一个

我已经使用以下命令打印张量列表:

for op in tf.get_default_graph().get_operations():
print (str(op.values()))

从该列表中,我选择了:  input_tensor =

sess.graph.get_tensor_by_name(“ linear / concat:0” )         input_tensor.set_shape([1,4])

但是形状不对应,我想这是因为线性分类器作为DNN起作用的原因,但是我怎么知道要选择哪个输入张量?

0 个答案:

没有答案