TensorFlowでnanを検知

TensorFlowでなんか結果がおかしいと思ってよく調べたら、nanが発生。
まさかのゼロ割発生。
以下のようにチェックコードを入れておけばnanの発生をお知らせしてくれる。

import tensorflow as tf
tf.debugging.enable_check_numerics()

パフォーマンスは落ちるかもしれないので、ガッツリ計算したい場合は外したほうがよさそうだけど、できるだけ入れておいた方がいいかも。