| | Layer (operation) | Input format | Output format |
| | Convolution layer 1 | [None,8,8,1] | [None,8,8,32] | | Batch normalization | [None,8,8,32] | [None,8,8,32] | | Nonlinear activation | [None,8,8,32] | [None,8,8,32] | | Pool layer | [None,8,8,32] | [None,4,4,32] | | Convolution layer 2 | [None,4,4,32] | [None,4,4,64] | | Batch normalization | [None,4,4,64] | [None,4,4,64] | | Nonlinear activation | [None,4,4,64] | [None,4,4,64] | | Pool layer 2 | [None,4,4,64] | [None,2,2,64] | | Fully connected layer | [None,256] | [None,6] | | Softmax | [None,6] | [None,6] |
|
|