eye = torch.eye(8).bool()