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

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

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

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

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

總結

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

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