冷凍Keras K.function（）tensorflow圖

Question

我有一個在Keras中訓練過的模型，我使用它的中間層輸出，使用K.function()。有沒有辦法將這個對象保存爲張量流圖？我想在張量流服務中使用這個對象，但我沒有看到凍結Keras的方法K.function()對象

Answer 1

我最終發現它，認爲我應該在這裏發佈它，以防將來任何人需要這樣做。我必須從K.function()對象中獲取節點名稱。

encoder = K.function([...]) # define your Function object 
encoder_tensor_names = [t.name for t in encoder.outputs] 
encoder_node_names = [tn.replace(':0', '') for tn in encoder_tensor_names] # node names are tensor names without :0 
graph_def = tf.graph_util.convert_variables_to_constants(
    K.get_session(), 
    K.get_session().graph.as_graph_def(), 
    encoder_node_names 
) 

然後按照代碼https://stackoverflow.com/a/44044405/5453184寫出來的二進制文件（或文本）文件。它的優點在於tensorflow足夠聰明，只保留前饋操作所需的圖形部分，在我的情況下，它也凍結了字嵌入，所以我不再需要單獨的單詞矢量文件。