我正在尝试将线性分类器导出为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的工作方式不同，但是我怎么知道选择哪个输入张量？

tensorflow tensorflow-lite