Hi,
Unfortunately, you won’t be able to give a batch of weights to the conv function as it does not support it.
But you can use the groups parameter. In particular, if your groups = n_channels, then each channel will be only multiplied with one weight. Does that match what you want?