Hello,
I wanted to write a variant of PReLU with one main distinction: it is always trainable for each channel-wise parameter instead of by layer.
The official PReLU implementation requires the user to enter the nchannels if it is not layer by layer.
Is there a way to discern that information given just the input tensor?