MemTorch
MemTorch copied to clipboard
Add Full Support for the `groups` Argument for Convolutional Layers
Currently, all convolutional layer types support groups=1 and groups=in_channels operation modes. Support should be added for groups=2, where the operation becomes equivalent to having two conv layers side by side, each seeing half the input channels and producing half the output channels, and both subsequently concatenated.