tensorflow中tensor与数组之间的转换

2024-10-11 22:53:17

# 主要是两个方法：

# 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)

最新文章

热门文章