Я пытаюсь экспортировать линейный классификатор в формат 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, но тогда как я могу узнать, какой входной тензор выбрать?