檢查Tensorflow中的NaN

Question

我想檢查tensorflow變量，如果它是NaN，則將其設置爲零。

我該怎麼做？下面的技巧似乎不起作用：

if tf.is_nan(v) is True: 
    v = 0.0 

Answer 1

如果v是0D張量，你可以使用tf.where測試和更新值：

import numpy as np 

v = tf.constant(np.nan)     # initialize a variable as nan ​ 
v = tf.where(tf.is_nan(v), 0., v) 
​ 
with tf.Session() as sess:  
    print(sess.run(v)) 

# 0.0 

Answer 2

你可以結合使用tf.is_nan與tf.cond如果tensorflow值NAN更改值。

Answer 3

圖書館像numpy的（在這種情況下，tensorflow）往往有自己的布爾實現，比較自定義的布爾類型的內存地址，CPython中內置的使用is會導致異常行爲。

要麼只檢查隱式布爾值 - >if tf.is_nan(v)或做一個相等比較if tf.is_nan(v) == True。

Answer 4

我希望這可以幫助你。 math.is_nan

import math 
if math.isnan(float(v)): 
    v = 0.0