Tensorflow Estimator：如何在model_fn中運行一個操作

Question

我得到了一個很短的問題，這個問題可能有一個非常簡單的答案，但我無法弄清楚，雖然我現在試了幾個小時。

我正在使用Tensorflow Estimator，我想訪問我的model_fn中的全局步驟。我試過tf.train.get_global_step，它給我一個張量。我需要global_step作爲一個整數雖然（或作爲一個字符串）！

所以我試圖eval（）函數（= tf.get_default_session（）。運行（T）），但它不工作..

乾杯！

Answer 1

您可以使用tf.cast將Tensor轉換爲int或string。

例如，

tf.cast(tf.train.get_global_step(), dtype=tf.int) 

見參考here。