溫馨提示×

溫馨提示×

您好，登錄后才能下訂單哦！

密碼登錄×
登錄注冊×
其他方式登錄
點擊 登錄注冊 即表示同意《億速云用戶服務條款》

tensor和numpy的互相轉(zhuǎn)換的實現(xiàn)示例

發(fā)布時間：2020-08-28 06:15:40 來源：腳本之家 閱讀：286 作者：L~花海 欄目：開發(fā)技術(shù)

要對tensor進行操作，需要先啟動一個Session,否則，我們無法對一個tensor比如一個tensor常量重新賦值或是做一些判斷操作，所以如果將它轉(zhuǎn)化為numpy數(shù)組就好處理了。下面一個小程序講述了將tensor轉(zhuǎn)化為numpy數(shù)組，以及又重新還原為tensor：

import tensorflow as tf
img1 = tf.constant(value=[[[[1],[2],[3],[4]],[[1],[2],[3],[4]],[[1],[2],[3],[4]],[[1],[2],[3],[4]]]],dtype=tf.float32)
img2 = tf.constant(value=[[[[1],[1],[1],[1]],[[1],[1],[1],[1]],[[1],[1],[1],[1]],[[1],[1],[1],[1]]]],dtype=tf.float32)
img = tf.concat(values=[img1,img2],axis=3)
sess=tf.Session()
#sess.run(tf.initialize_all_variables())
sess.run(tf.global_variables_initializer())
print("out1=",type(img))
#轉(zhuǎn)化為numpy數(shù)組
img_numpy=img.eval(session=sess)
print("out2=",type(img_numpy))
#轉(zhuǎn)化為tensor
img_tensor= tf.convert_to_tensor(img_numpy)
print("out2=",type(img_tensor))

輸出：

out1= <class 'tensorflow.python.framework.ops.Tensor'>
out2= <class 'numpy.ndarray'>
out2= <class 'tensorflow.python.framework.ops.Tensor'>

以上就是本文的全部內(nèi)容，希望對大家的學習有所幫助，也希望大家多多支持億速云。

向AI問一下細節(jié)

免責聲明：本站發(fā)布的內(nèi)容（圖片、視頻和文字）以原創(chuàng)、轉(zhuǎn)載和分享為主，文章觀點不代表本網(wǎng)站立場，如果涉及侵權(quán)請聯(lián)系站長郵箱：is@yisu.com進行舉報，并提供相關(guān)證據(jù)，一經(jīng)查實，將立刻刪除涉嫌侵權(quán)內(nèi)容。

AI
助
手