인프런 커뮤니티 질문&답변

안녕하세요!

여기서는 Flatten하고 관계는 없습니다.

outputs 크기가 (배치 사이즈)x(클래스 수) 입니다. 즉, 각 이미지 당 클래스 수 만큼의 크기의 벡터로 표현 됩니다. 따라서 예측값을 정할 때 각 벡터마다 가장 큰 값을 기준으로 예측값을 정하기 때문에 열 기준으로 큰 값을 산정합니다.

예를 들어 클래스가 3개짜리인 문제에 배치 크기가 2인 outputs= [[0.1, 0.4, 0.5],[0.5 ,0.3, 0.2]] 라고 하면 각 열 기준 첫번째 이미지의 예측값은 2이고 두번째 예측값은 0이 되는 것입니다 :)