@tf.function

텐서 선언하기 a = tf.constant(2) # 텐서를 선언합니다. print(tf.rank(a)) # 텐서의 랭크를 계산합니다. b = tf.constant([1, 2]) print(tf.rank(b)) c = tf.constant([[1, 2], [3, 4]]) print(tf.rank(c)) 결과 tf.Tensor(0, shape=(), dtype=int32) tf.Tensor(1, shape=(), dtype=int32) tf.Tensor(2, shape=(), dtype=int32) 즉시 실행모드를 통한 연산 텐서플로우 1.x에서는 계산 그래프를 선언, 초기화 뒤 세션을 통해 값을 흐르게 하는 등의 많은 작업을 필요. 2.x의 버전에서는 텐서를 복잡한 과정 없이 파이썬처럼 사용할 수 있음. ..
Codezoy
'@tf.function' 태그의 글 목록