Separate the linear layer for different features

[benoriol (Ben)]

Your’re right, the parameters in the convolution would be shared… My bad