더북(TheBook)

3.5.2 텐서 연산: 텐서플로에서 수학 계산하기

넘파이와 마찬가지로 텐서플로는 수학 공식을 표현하기 위해 많은 텐서 연산을 제공합니다. 다음은 몇 가지 예입니다.

코드 3-9 기본적인 수학 연산

a = tf.ones((2, 2))
b = tf.square(a) 
c = tf.sqrt(a) 
d = b + c 
e = tf.matmul(a, b) 
e *= d 

제곱을 계산합니다.

제곱근을 계산합니다.

두 텐서를 더합니다(원소별 연산).

두 텐서의 점곱을 계산합니다(2장에서 보았습니다).

두 텐서를 곱합니다(원소별 연산).

중요한 점은 앞의 연산이 모두 바로 실행된다는 것입니다. 넘파이처럼 언제든지 현재 결괏값을 출력할 수 있습니다. 이를 즉시 실행(eager execution) 모드라고 부릅니다.

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