Tensorflow Estimator: Wie eine Operation innerhalb model_fn

Question

ich eine sehr kurze Frage bekam laufen, das ist wahrscheinlich eine sehr einfache Antwort hat, aber ich kann es einfach nicht herausfinden, obwohl ich jetzt stundenlang versucht.

Ich verwende Tensorflow Estimator und ich möchte den globalen Schritt in meinem model_fn zuzugreifen. Ich habe tf.train.get_global_step versucht, was mir einen Tensor zurückgibt. Ich brauche den global_step aber als Integer (oder als String)!

Also habe ich versucht, eval() (= tf.get_default_session(). Laufen (t)), aber es funktioniert nicht ..

Prost!

Answer 1

Sie können tf.cast verwenden, um die Tensor zu int oder string zu werfen.

Zum Beispiel

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

Siehe Referenz here.