我正在尝试将线性分类器导出为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起作用的原因,但是我怎么知道要选择哪个输入张量?