擅长:python、mysql、java
<p>我想你需要用tf.select。</p>
<pre><code>w = tf.select(tf.is_nan(w), tf.ones_like(w) * NUMBER, w); #if w is nan use 1 * NUMBER else use element in w
</code></pre>
<hr/>
<p><strong>更新:</strong>TensorFlow 1.0有<a href="https://github.com/tensorflow/tensorflow/issues/6899" rel="noreferrer">deprecated</a><code>tf.select</code>支持Numpy兼容<a href="https://www.tensorflow.org/api_docs/python/tf/where" rel="noreferrer">^{<cd2>}</a>。</p>