当使用Tensorflow与Python绑定时,如何将一个张量转换为numpy数组?


当前回答

TensorFlow 1.倍

文件夹特遣部队。1、只需使用以下命令:

a = tf.constant([[1, 2], [3, 4]])                 
b = tf.add(a, 1)
out = tf.multiply(a, b)
out.eval(session=tf.Session())

输出将是:

# array([[ 2,  6],
#       [12, 20]], dtype=int32)

其他回答

如果你看到有一个方法_numpy(), 例如,对于一个EagerTensor,简单地调用上面的方法,你将得到一个ndarray。

TensorFlow 1.倍

文件夹特遣部队。1、只需使用以下命令:

a = tf.constant([[1, 2], [3, 4]])                 
b = tf.add(a, 1)
out = tf.multiply(a, b)
out.eval(session=tf.Session())

输出将是:

# array([[ 2,  6],
#       [12, 20]], dtype=int32)

要将张量转换回numpy数组,只需在转换后的张量上运行.eval()。

你可以用以下方法将tensorflow中的张量转换为numpy数组。

第一: 使用np.array (your_tensor)

第二: 使用your_tensor.numpy

我成功地把TensorGPU变成了np。数组使用以下 :

np.array(tensor_gpu.as_cpu())

(直接使用TensorGPU只会导致包含TensorGPU的单元素数组)。