Thanks for the reply. But I don’t think the weight should be shared between features, which is the case of the convolution.
Rather, I’m thinking about using torch.nn.ModuleList, but I’m wondering if there is a more elegant way to do that.