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>
注意:
以上是如何将 TensorFlow 张量转换为 NumPy 数组?的详细内容。更多信息请关注PHP中文网其他相关文章!