Autograd `Function` for conv layers

[ptrblck]

Have a look at this post which gives more details on the backward pass in conv layers.