1. 程式人生 > >如何檢視張量tensor,並將其轉換為numpy資料

如何檢視張量tensor,並將其轉換為numpy資料

在tensorflow 中一般資料都是用tensor來表示,而在python 中一般是用numpy包,然而有時候需要列印變數的資料,可用以下方法來列印:

一、

import tensorflow as tf
a = tf.constant(2.1) #定義tensor常量
with tf.Session() as sess:
    print (sess.run(a))

二、

還可以通過.eval函式可以把tensor轉化為numpy類資料,程式如下:

import tensorflow as tf
a = tf.constant(2.1) #定義tensor常量
sess=tf.Session()
b=a.eval(session=sess)
print (b)
print (type(b))

三、

此外,還可以通過tf.convert_to_tensor函式可以把numpy轉化為tensor 類資料:

import tensorflow as tf
a = tf.constant(2.1) #定義tensor常量
sess=tf.Session()
b=a.eval(session=sess)
b=tf.convert_to_tensor(a)
print (b)