在tensorflow中numpy与tensor的互相转化
在进行tensorflow运算时,我们常常需要进行tensor与numpy的转换。
我们用tensor=tf.convert_to_tensor(numpy)实现numpy转换到tensor
用sess.run()转换tensor到numpy
import tensorflow as tf
import numpy as np
a=np.array([[1,2,3,4],[5,6,7,8]])
b=np.array([[1,2,3,4],[5,6,7,8]])
a=tf.convert_to_tensor(a,dtype=tf.float32)
b=tf.convert_to_tensor(b,dtype=tf.float32)
c=tf.add(a,b)
with tf.Session() as sess:
print(sess.run(c))
sess.close()
运行结果