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

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

總結(jié)

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

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。