tf.constant¶

tf.constant 함수는 상수 텐서를 생성합니다.

예제1¶

import tensorflow as tf
import numpy as np

a = tf.constant([1, 2, 3, 4])
b = tf.constant(np.array([1, 2, 3, 4]))

print(a)
print(b)

tf.Tensor([1 2 3 4], shape=(4,), dtype=int32)
tf.Tensor([1 2 3 4], shape=(4,), dtype=int64)

tf.constant 함수는 파이썬 리스트 또는 NumPy 어레이로부터 상수 텐서를 생성합니다.

import tensorflow as tf
import numpy as np

a = tf.constant([1, 2, 3, 4], dtype=tf.float64)
b = tf.constant(0, shape=(2, 3))

print(a)
print(b)

tf.Tensor([1. 2. 3. 4.], shape=(4,), dtype=float64)
tf.Tensor(
[[0 0 0]
 [0 0 0]], shape=(2, 3), dtype=int32)

tf.constant 함수의 dtype은 생성될 값의 자료형을 지정합니다.

shape을 지정하면 입력값을 지정한 형태로 생성합니다.