[TensorFlow] Tensorflow Dataset - 'Tensor' object has no attribute 'numpy' 에러 해결

tf.Tensor 객체에 .numpy() 메소드를 호출 했을 때, 'Tensor' object has no attribute 'numpy' 메세지를 볼 때가 있다.

코드는 아래와 같다.

import tensorflow as tf
import tensorflow_datasets as tfds
 
datasets, ds_info = tfds.load(name='mnist', with_info=True, as_supervised=True)
mnist_train, mnist_test = datasets['train'], datasets['test']
 
for mnist_example in mnist_train.take(1):
    image = mnist_example[0]
    image.numpy()
    break

 

9번째 라인에서 image.numpy()를 호출했는데, 'Tensor' object has no attribute 'numpy' 에러가 뜬다. 이럴 때는 어떻게 해야 할까?

텐서플로를 import 하면서 바로 아래에 이렇게 즉시 실행 코드를 넣으면 된다.

import tensorflow as tf
tf.enable_eager_execution()

정확한 원인은 모르겠으나, 텐서플로에 아래 링크와 같이 즉시 실행 모드가 있는 것으로 보인다.

https://www.tensorflow.org/tutorials/eager/eager_basics

즉시 실행 기초  |  TensorFlow Core

문서 초반에 이런 문구가 있다.

"시작하기 위해서 텐서플로 모듈을 임포트하고 즉시 실행(eager execution)을 활성화합니다. 즉시 실행 활성화로 텐서플로를 조금 더 대화형 프론트엔드(frontend)에 가깝게 만들어 줍니다."

 

즉, tf.enable_eager_execution() 를 사용하면 tensor를 즉시 실행해주고, 아니라면 session을 사용해서 활성화를 해야하는 것으로 보인다.