Training a fully connected layer as a convolution layer