Vectorized nearest-neighbour classifier as a linear layer

Implement a 1-nearest-neighbour classifier in NumPy with no Python loops. Then rewrite it as the forward pass of a small network, a matrix product plus a bias followed by an activation, and explain why the two give the same predictions.