x = torch.randn(8, 2)