Tesnsorflow.cast 텐서 데이터 타입 변경하기

Tensorflow.cast => 텐서를 특정 dtype으로 변환

tf.cast(
    x, dtype, name=None
)

 

변환하는 dtype에는 tf.uint8, tf,uint16,tf.uint32,tf.float16 등 여러 데이터 타입을 설정할 수 있습니다.

 

 

주로 이미지 데이터를 다룰 때, 255.0으로 나누는 작업을 수행하기 위해, cast를 사용하는 것 같습니다.

tf.cast(image, tf.float32) / 255.0

 

Source:

https://www.tensorflow.org/api_docs/python/tf/cast