[We-Co] TensorFlow 형변환

안녕하세요.

위기의 코딩맨 입니다.

오늘은 TensorFlow의 데이터 타입의 형변환하는 방법을 알아보도록 하겠습니다.

 

[Cast]

x = tf.constant([1, 2, 3], name='x', dtype = tf.float32)

print(x.dtype)

 

x를 Float32의 타입으로 생성을 하고

타입을 출력해보면

 

결과 값

dtype이 설정한 float32로 출력이 되는 것을 확인할 수 있습니다.

 

 

 

이제 형 변환을 하기 위해 TensorFlow의 Cast 함수를 사용하여 변환 하도록 하겠습니다.

첫번째 인자로 변환할 데이터, 두번째 인자로는 변화를 진행할 Type을 설정해 주시면 됩니다.

 

x = tf.cast(x, tf.int64)
print(x.dtype)

 

float32 Type을 int64로 형 변환을 진행하여 출력해 보았습니다.

결과 값 2

x의 Type이 변환이 진행 된것을 확인할 수 있습니다.