Better way to expand along non singleton dimensions?

1

Is there a better way to accomplish this?

mask = torch.tensor([[1, 0], [0, 1]])
for i in range(6):
    mask = torch.cat((mask, torch.tensor([[1, 0], [0, 1]])))
2

repeat should yield this output:

x = torch.tensor([[1, 0], [0, 1]])
x = x.repeat(7, 1)