Pytorch conv2D groups

This post might be helpful.