Pytorch Conv2d Explained