TensorFlow 인쇄 tensor 값 의 실현 방법

최근 에는 TF 로 CNN 의 이미지 분 류 를 해 왔 는데,소프트 맥 스 층 이 예측 결 과 를 얻 으 면 태그 와 비교 할 수 있 도록 예측 결 과 를 보고 싶다.특별히 이것 을 보충 하여 자신 이 기억 할 수 있 도록 하 다.
저 는 지금 softmax 층 을 통 해 변 수 를 얻 었 습 니 다 trainlogits,내 가 직접 print(trainlogits)에서 얻 은 결 과 는 다음 과 같다.

아무것도 모 르 는 것 같 습 니 다.
사실 tensor flow 는 출력 중간 값 방법 을 제공 하여 debug 에 편리 합 니 다.
이 함수 가 바로[tf.Print]입 니 다.

tf.Print( 
input_, 
data, 
message=None, 
first_n=None, 
summarize=None, 
name=None 
)
인자:
input_: 이 op 을 통 해 tensor.
data:이 op 이 계 산 된 후에 출력 된 tensor list 를 인쇄 합 니 다.
message:오류 메시지 의 접 두 사 는 string 입 니 다.
first_n:first 만 기록n 회.항상 마이너스 기록 하기;이것 은 결 성 이다.
summarize:모든 tensor 에 인쇄 된 항목 의 수 입 니 다.None 이 라면,tensor 를 입력 할 때마다 세 개의 요소 만 인쇄 합 니 다.
name:op 의 이름.
반환 값:
input같은 tensor.
print(trainlogits)sess.run(tf.Print(trainlogits,[train_logits])후 얻 은 결 과 는 다음 그림 과 같다.

값 을 인쇄 할 수 있 음 을 발 견 했 습 니 다!하지만 3 개의 항목 만 출력 할 수 있 습 니 다.인자 summarize 는 기본적으로 3 개의 인자 만 인쇄 하기 때 문 입 니 다!
sess.run(tf.Print(trainlogits,[train_logits])를 sess.run(tf.Print(trainlogits,[train_logits],summarize=134)후(나 는 134 가지 분류)얻 은 결 과 는 다음 과 같다.

이렇게 하면 tensor 의 중간 결 과 를 성공 적 으로 인쇄 할 수 있 습 니 다!앞으로 인삼 을 조절 하고 모델 효 과 를 보면 훨씬 직관 적 입 니 다!
이 텐 서 플 로 우 가 텐 서 값 을 인쇄 하 는 실현 방법 은 바로 편집장 이 여러분 에 게 공유 하 는 모든 내용 입 니 다.참고 하 실 수 있 고 많은 응원 부 탁 드 리 겠 습 니 다.

좋은 웹페이지 즐겨찾기