반응형
Deep Learning & Machine Learning/강좌&예제 코드2020. 8. 1. 18:40Tensorflow 예제 - tf.argmax 함수 사용법

저수준 API로 작성된 MNIST 코드에서 사용하는 tf.argmax 함수에 대해 살펴봅니다. 2018. 8.29 최초작성 2020. 8. 1 내용 확인 및 Tensorflow 2.x에 맞게 수정 tf.argmax 두번째 인자값의 범위는 [-rank(input), rank(input))로 한정되어 있습니다. 텐서플로우에서 rank는 텐서의 원소 하나에 접근하기 위해 필요한 인덱스의 개수입니다. 1차원 배열의 경우 최대 인덱스 개수는 1(=rank가 1 ) 이기 때문에 두번째 인자로 0만 사용할 수 있습니다.(음의 범위는 논외로 합니다. ) 한 방향으로 (1차원의 경우 열,행 구분이 없습니다.) 최대값을 찾아 인덱스 값을 찾을 수 있습니다. import tensorflow as tf a = tf.const..

반응형
image