@ptrblck
Since groups == channels is faster than group convolution and basic convolution, I want to know something else.
If I set out_channels == groups == channels, it becomes a depthwise convolution. Does depthwise convolution is also highly optimized, or just because of the reduction of computation?