如何将 TensorFlow 张量转换为 NumPy 数组？

在 TensorFlow 中将张量转换为 NumPy 数组

TensorFlow 提供了一种方便的方法 .numpy()，可以将张量转换为 NumPy 数组。

TensorFlow 2.x

启用急切执行使 TensorFlow 操作立即可执行，允许您直接在张量上调用 .numpy()：

<code class="python">import tensorflow as tf

a = tf.constant([[1, 2], [3, 4]])
b = tf.add(a, 1)

print(a.numpy())
# array([[1, 2],
#        [3, 4]], dtype=int32)

print(b.numpy())
# array([[2, 3],
#        [4, 5]], dtype=int32)</code>

TensorFlow 1 .x

如果 TensorFlow 1.x 中禁用了 eager execution，您可以创建一个图并执行它来获取 NumPy 数组：

<code class="python">a = tf.constant([[1, 2], [3, 4]])
b = tf.add(a, 1)
out = tf.multiply(a, b)

with tf.compat.v1.Session() as sess:
    print(sess.run(out))
# array([[ 2,  6],
#        [12, 20]], dtype=int32)</code>

注意：

• .numpy() 可能会与原始张量共享内存，因此对一个张量的更改可能会影响另一个。
• 如果遇到错误“AttributeError: 'Tensor'对象没有属性“numpy”，”确保 TF 2.0 已正确安装，或启用急切执行。

以上是如何将 TensorFlow 张量转换为 NumPy 数组？的详细内容。更多信息请关注PHP中文网其他相关文章！