tensorflow 中tensor與數組之間的轉換


# 主要是兩個方法:
# 1.數組轉tensor:數組a,  tensor_a=tf.convert_to_tensor(a)
# 2.tensor轉數組:tensor b, array_b=b.eval()
#
# 下面看一個例子

import tensorflow as tf
import numpy as np

a=np.array([[1,2,3],[4,5,6],[7,8,9]])
print (a)
b=tf.constant(a)

with tf.Session() as sess:
    print (b)
    for x in b.eval():      #b.eval()就得到tensor的數組形式
        print (x)

    print ('a是數組',a)

    tensor_a=tf.convert_to_tensor(a)
    print ('現在轉換爲tensor了...',tensor_a)

下面是程序運行結果:


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章