Tenor 和numpy array 相互转换

a = np.array([1,2,3])
b = tf.constant([1,2,3])

numpy array 转 Tensor

res = tf.convert_to_tensor(a)

Tensor 转 numpy array

res = b.eval(session=sess)

二者的转换实际上就是静态图阶段的数据运行时的数据之间的转换
其实sess.run(tensor)和tensor.eval(session=sess)效果一样,但是tensor.eval()写起来更高效快捷

0%