1.tf.reshape() #調(diào)整tensor的形狀

img = .... img_res = tf.reshape(img, [-1])

著重記錄是shape=[-1]時(shí)，相當(dāng)于flatten操作，與此對(duì)應(yīng)的是在tf.placeholder()中shape=None時(shí)，要求輸入為一維。
2.tf.convert_to_tensor() #將ndarray等轉(zhuǎn)換為tensor

x = np.array([[1,1],[2,2]]) tx = tf.convert_to_tensor(x)

3.tf.where()#有兩種用法，分開(kāi)討論
? ?（1）tf.where(tensor)，#tensor 為一個(gè)bool 型張量，where函數(shù)將返回其中為true的元素的索引
? ?（2）tf.where(tensor, a?, b), #根據(jù)tensor的對(duì)應(yīng)位置的bool值來(lái)選擇a或b中對(duì)應(yīng)位置的元素，true，從a中選擇，false，從b中選擇。該函數(shù)在語(yǔ)義分割等任務(wù)中常用。注（tensor, a , b 的shape一致）

總結(jié)

以上是生活随笔為你收集整理的tensorflow 之 最近用到的几个小操作tf.reshape，tf.convert_to_tensor，tf.where的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

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