더북(TheBook)

3.5.1 상수 텐서와 변수

텐서플로에서 어떤 작업을 하려면 텐서가 필요합니다. 텐서를 만들려면 초깃값이 필요합니다. 예를 들어 모두 1이거나 0인 텐서를 만들거나(코드 3-1), 랜덤한 분포에서 뽑은 값으로 텐서를 만들 수 있습니다(코드 3-2).

코드 3-1 모두 1 또는 0인 텐서

>>> import tensorflow as tf
>>> x = tf.ones(shape=(2, 1)) 
>>> print(x)
tf.Tensor(
[[1.]
 [1.]], shape=(2, 1), dtype=float32)
>>> x = tf.zeros(shape=(2, 1)) 
>>> print(x)
tf.Tensor(
[[0.]
 [0.]], shape=(2, 1), dtype=float32)

np.ones(shape=(2, 1))과 동일합니다.

np.zeros(shape=(2, 1))과 동일합니다.

신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.