훈련시킨 후 테스트 해보기

 

테스트 한 후에 아무 입력이나 넣어봤는데, 계속 에러. 왜 그런가 했더니 network 클래스에서 받는 배렬의 차원이 안맞았음. 그래서 요 예제에 한해서 반드시 저렇게 reshape를 해서 4차원으로 만들어줘야 함. 4차원이라고 해서 4차원에 뭔가 꽉찬 데이타는 아님. 그냥 28x28 픽셀 데이타인데, 그걸 형식만 4차원으로 하라는 뜻임. 따라서, arr[1]은 당연히 없음.