We-Co

[We-Co] TensorFlow Linspace 본문

Python/Tensorflow

[We-Co] TensorFlow Linspace

위기의코딩맨 2021. 7. 18. 19:45
반응형

안녕하세요.

위기의 코딩맨입니다.

오늘은 TensorFlow의 Linspace에 대해 알아보도록 하겠습니다.

간단하게 미리 설명하면 주어진 축으로 일정한 값을 생성해 나가는 것입니다.

 

c = tf.linspace(a, b, c)

 

소스를 확인하면  a부터 b까지 c개 만큼 생성해주세요. 라고 생각하면 된다.

예제를 작성해보면

 

with tf.Session() as sess:
    c = tf.linspace(0.0, 4.0,5)
    print(c.eval())

 

with으로 섹션을 묶어 생성해주고

0부터 4까지 5개의 값을 생성해서 값을 출력하는 소스입니다.

출력 값

0,1,2,3,4 총 5가지의 출력 값이 존재하다는 것을 알 수 있다.

여기서 eval()의 함수는 세션 객체의 값을 볼 수 있는 함수이다.

중요한 점은 그 해당 객체가 메인으로 잡혀 있어야 그 값을 확인 할 수 있습니다. 

예제에서는 with으로 생성한 Session이 메인으로 잡혀 있어 c.eval()로 값을 확인이 가능했던 것이다.

with를 자주 사용하자..!

만약 with으로 잡지 않고 Session을 따로 생성하게 되면 InteractiveSession() 함수를 사용하여

해당 Session을 설정해줘야 한다.

 

sess = tf.InteractiveSession()
c = tf.linspace(0.0, 4.0,5)
print(c.eval())

 

출력 값 2

설정을 해주지 않으면 Session을 default로 잡지 않았다는 에러를 반환하게 된다.

에러

 

오늘은 Linspace에 대해서 알아보고 그 속의 값을 출력하는 방법을 알아보았습니다.

TensorFlow를 알아갈 수록 재미가 붙어가고 있습니다.!

반응형