Snap interview question

Write code for a convolution neural network