tf.where() 详解
Posted K同学啊
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了tf.where() 详解相关的知识，希望对你有一定的参考价值。
tf.where(tensor) ：tensor 为一个bool 型张量，where函数将返回其中为true的元素的索引。
tf.where(tensor,a,b)：a,b为和tensor相同维度的tensor，将tensor中的true位置元素替换为ａ中对应位置元素，false的替换为b中对应位置元素。
以上是关于tf.where() 详解的主要内容，如果未能解决你的问题，请参考以下文章
TF报错：Use tf.where in 2.0, which has the same broadcast rule as np.where
TF报错：Use tf.where in 2.0, which has the same broadcast rule as np.where