2017-06-01 29 views
1

我正在TensorFlow中訓練神經網絡以用於Java應用程序。使用freeze_graph.py將圖形寫入pb文件。在TensorFlow中將佔位符轉換爲常量

但是,圖中有許多佔位符用於保持丟失層保持概率。我想在編寫圖時將這些轉換爲常量(= 1.0),因爲在Java應用程序中使用網絡時,我不需要擔心提供這些值。

是否可以將佔位符轉換爲常量?

回答

1

更接近你想要的東西是placeholder_with_default。這是一個佔位符,當不餵食時會採用默認值。

您既可以用想要的值填充它,也可以忘記它,並讓它將您指定的值作爲默認值。

相關問題