Locally connected 2D layer without summation over colors

I want to create a neural network with a locally connected layer but without summation over the 3rd dimension (colors) of the inputs.

I saw in the docs of LocallyConnected2D that there is no “group” argument as in the conv2d.

Is there a way to do that?

Do you mean something like this?

1 Like

Yes, I don’t see the solution there - you know how exactly tackle this problem?

I suggest you to comment on that ticket as It was just closed in june.

1 Like