tf.cast

tf.cast 함수는 텐서를 새로운 자료형으로 변환합니다.



예제1

import tensorflow as tf

x = tf.constant([1.8, 2.2], dtype=tf.float32)
y = tf.cast(x, tf.int32)

print(x)
print(y)

tf.Tensor([1.8 2.2], shape=(2,), dtype=float32)
tf.Tensor([1 2], shape=(2,), dtype=int32)

x는 (2,) 형태를 갖고, tf.float32 자료형을 갖는 텐서입니다.

tf.cast()를 사용해서 tf.int32 자료형을 갖도록 변환했습니다.



예제2

import tensorflow as tf

x = tf.constant([1.8, 2.2], dtype=tf.float32)
y = tf.dtypes.cast(x, tf.int32)

print(x)
print(y)

tf.Tensor([1.8 2.2], shape=(2,), dtype=float32)
tf.Tensor([1 2], shape=(2,), dtype=int32)

tf.dtypes.cast()는 tf.cast()의 alias입니다.

tf.dtypes.cast()를 사용해서 tf.int32 자료형을 갖도록 변환했습니다.



이전글/다음글

이전글 :

공유하기