I’m not sure how the mentioned shape of [3, 128] fits the indices in [0, 10] from the description.
However, if you want to concatenate tensors, you could use torch.cat or torch.stack, if you want to add a new dimension.