Add a hook to add some tensors to the summary

that's nice :-)

thanks

This is only useful for keras, no? I don't see how this can be used for estimators since the tensors are created in the model_fn

Yes I wrote it for Keras Models.