如何將 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中文網其他相關文章！