[Tensorflow / Keras] Tf.cast 함수

tf.cast함수에 대해 알아본다.

tf.cast함수는 float형을 int형으로, 혹은 boolean형을 int형으로 캐스팅할 때 사용한다.

예시를 통해 알아가보자

float to int

import tensorflow as tf
import numpy as np

float_n = np.random.random()
print(float_n)

int_n = tf.cast(float_n, tf.int64)
print(int_n)

 

boolean to int

import tensorflow as tf
import numpy as np

bool_n = True
print("bool_n is {}".format(bool_n))
int_n = tf.cast(bool_n, tf.int64)
print("bool_n casted is {}".format(int_n))

import tensorflow as tf
import numpy as np

bool_n = False
print("bool_n is {}".format(bool_n))
int_n = tf.cast(bool_n, tf.int64)
print("bool_n casted is {}".format(int_n))

 

이런식으로 데이터 타입이 변경되는 것을 알 수 있다.