텐서플로 tf.cast 함수 사용법

tf.cast 함수는 보통 조건에 따른 True, False 의 판단 기준에 따라 True 면 1, False 면 0을 반환한다.

즉, 

tf.cast("조건")

조건에 따라 1 또는 0 이 반환된다.

ex) x라는 변수값이 0.5 이상이면 1, 0.5 미만이면 0을 반환하고 싶을때

import tensorflow as tf

x = tf.constant([-1, 3.5], tf.float32)

c = tf.cast(x>0.5, tf.float32)

sess = tf.Session()

y = sess.run(c)

print(y)

> [ 0. 1. ]