You don’t need to slice and concatenate x, as nn.Linear layers will take an input of [batch_size, *, in_features] and apply the linear layer on all additional dimensions specified as *.
However, I’m still unsure about your picture.
The dots represent the activations, while the lines should represent the weight matrices (or the linear layers in general).
What would the first and last lines represent?