tf.where(tensor) ：tensor 為一個bool 型張量，where函數將返回其中為true的元素的索引。

tf.where(tensor,a,b)：a,b為和tensor相同維度的tensor，將tensor中的true位置元素替換為ａ中對應位置元素，false的替換為b中對應位置元素。

總結

以上是生活随笔為你收集整理的tf.where() 详解的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。