首頁 > 後端開發 > Python教學 > 如何將 Tensorflow 張量轉換為 NumPy 陣列?

如何將 Tensorflow 張量轉換為 NumPy 陣列?

Barbara Streisand
發布: 2024-11-03 21:23:03
原創
783 人瀏覽過

How do I convert a Tensorflow Tensor to a NumPy Array?

在 Tensorflow 中將張量轉換為 Numpy 數組

在 Tensorflow 中,將張量轉換為 numpy 數組非常簡單。操作方法如下:

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

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

a_numpy = a.numpy()  # Convert tensor 'a' to numpy array
b_numpy = b.numpy()  # Convert tensor 'b' to numpy array

print(a_numpy)
# [[1 2]
#  [3 4]]
print(b_numpy)
# [[2 3]
#  [4 5]]</code>
登入後複製

注意:

  • 在 Tensorflow 2.x 中,預設啟用急切執行。因此,您只需在張量物件上呼叫 .numpy() 即可。
  • 如果您收到有關 .numpy() 屬性的 AttributeError,請確保正確安裝 TF 2.0 並啟用 Eager Execution。
  • Numpy 陣列與其對應的張量共享記憶體。對其中一個所做的更改會反映在另一個中。
  • 如果停用 Eager Execution,您可以建立一個圖表並使用 tf.compat.v1.Session 來執行它。

以上是如何將 Tensorflow 張量轉換為 NumPy 陣列?的詳細內容。更多資訊請關注PHP中文網其他相關文章!

來源:php.cn
本網站聲明
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn
作者最新文章
熱門教學
更多>
最新下載
更多>
網站特效
網站源碼
網站素材
前端模板