TensorFlow Tensor를 NumPy 배열로 어떻게 변환합니까?

TensorFlow에서 Tensor를 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에서 즉시 실행이 비활성화된 경우 그래프를 생성하고 실행하여 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 Tensor를 NumPy 배열로 어떻게 변환합니까?의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!