Tensorflow에서 표현식의 출력 이름을 지정하십시오.

Question

특정 표현식의 출력에 이름을 지정하여 코드의 다른 부분에서 그래프에서 검색 할 수 있는지 궁금합니다. 나는 전체 기능의 계층 구조를 통해이를 반환하지 않고 다른 위치에 c을 당겨 수 있다면 좋을 텐데 디버그 목적을 위해 예를

def A(a, b): 
    c = a + b 
    d = d * a 
return d 

를 들어

.

아이디어가 있으십니까?

미리 감사드립니다.

Answer 1

저는 a이고 b은 텐서입니다.

는 어느 쪽이든 당신은 tf.identity

def A(a, b): 
    c = a + b 
    c = tf.identity(c, name = "c") 
    d = d * a 
return d 

은 어느 대신 +의 tf.add 동작을 사용하여 c에 이름 부여 :

def A(a, b): 
    tf.add(a, b, name = "c") 
    d = d * a 
return d 

어느 쪽이든, 당신은 tf.get_variable('c:0')와 c를 검색 할 수를 (당신이해야 할 수도 있습니다 정확한 범위로).