Setting up unique custom filters in Conv2D for each image?


I know how to set up a custom filter for conv2D but, How do i set up unique filters for each image in my batch. For example if I have 10 images in my batch, and let’s say I want to apply k filters of size 3 x 3 to each image, Now I want to be able to feed unique K filters to each of the 10 images something like 10 x k x num_channels x 3 x 3. How do I do that?

Just define 10 different convolutions in the main nn.Module and redirect each sample to each convolution. Later on collect them and concatenate them back.