首页 > 后端开发 > Python教程 > 如何将 TensorFlow 张量转换为 NumPy 数组?

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

Linda Hamilton
发布: 2024-11-03 19:49:03
原创
753 人浏览过

How do I Convert TensorFlow Tensors to NumPy Arrays?

在 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中文网其他相关文章!

来源:php.cn
本站声明
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
作者最新文章
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板