If you want to stack these tensors together, you could use something like this:
a = torch.randn(128)
b = torch.randn(128)
c = torch.stack((a, b)) * torch.randperm(2).unsqueeze(1)

which would zero out one of these tensors completely.
If that’s not your use case, could you explain the shapes a bit more?