Rumah  >  Artikel  >  pembangunan bahagian belakang  >  Bagaimanakah saya boleh Menukar Tensor TensorFlow kepada Array NumPy?

Bagaimanakah saya boleh Menukar Tensor TensorFlow kepada Array NumPy?

Linda Hamilton
Linda Hamiltonasal
2024-11-03 19:49:03692semak imbas

How do I Convert TensorFlow Tensors to NumPy Arrays?

Menukar Tensor kepada NumPy Array dalam TensorFlow

TensorFlow menyediakan kaedah yang mudah, .numpy(), untuk menukar tensor kepada tatasusunan NumPy.

TensorFlow 2.x

Mendayakan pelaksanaan yang bersemangat menjadikan operasi TensorFlow serta-merta boleh dilaksanakan, membolehkan anda memanggil .numpy() terus pada tensor:

<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

Jika pelaksanaan bersemangat dilumpuhkan dalam TensorFlow 1.x, anda boleh mencipta graf dan melaksanakannya untuk mendapatkan tatasusunan 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>

Nota:

  • .numpy() mungkin berkongsi memori dengan tensor asal, jadi perubahan kepada satu boleh menjejaskan yang lain.
  • Jika anda menghadapi ralat "AttributeError: 'Tensor' objek tidak mempunyai atribut 'numpy'," pastikan TF 2.0 dipasang dengan betul atau dayakan pelaksanaan yang bersemangat.

Atas ialah kandungan terperinci Bagaimanakah saya boleh Menukar Tensor TensorFlow kepada Array NumPy?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!

Kenyataan:
Kandungan artikel ini disumbangkan secara sukarela oleh netizen, dan hak cipta adalah milik pengarang asal. Laman web ini tidak memikul tanggungjawab undang-undang yang sepadan. Jika anda menemui sebarang kandungan yang disyaki plagiarisme atau pelanggaran, sila hubungi admin@php.cn