Firing inhibition only works for flattened layers

[jeshraghian]

Description

the inhibition argument of neurons is only setup for single dimensional channels / outputs of nn.Linear
It does not work for convolutional layers.

The way I see it, there are two options:

1. Only allow 1 neuron to fire from an entire layer; this is how it is currently setup for Linear layers
2. Allow 1 neuron to fire per-channel. Trickier, as the channel dimension must be known.

[pengzhouzp]

The second option makes more sense to me, as features are extracted by different filters/channels. What is the meaning of the channel dimension must be known? Maybe I need to see the code to understand haha.