eye = torch.eye(8)