AlexNet의 첫 번째 합성곱층 커널의 크기는 11×11×3이며, 스트라이드를 4로 적용하고 특성 맵을 96개 생성하여 55×55×96의 출력을 갖습니다. 첫 번째 계층을 거치면서 GPU-1에서는 주로 컬러와 상관없는 정보를 추출하기 위한 커널이 학습되고, GPU-2에서는 주로 컬러와 관련된 정보를 추출하기 위한 커널이 학습됩니다.

각 GPU의 결과가 다음 그림과 같이 나타납니다.

▲ 그림 6-10 AlexNet GPU-1·2 적용 결과

이제 텐서플로 2 코드로 AlexNet에 대해 살펴봅시다. 먼저 필요한 라이브러리를 호출합니다. 전반적인 코드는 LeNet-5와 동일하며 네트워크를 구현하는 부분만 다릅니다.