From b108aef212afc9b37ece29bea44f35b8686aac02 Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Sun, 22 Oct 2023 17:08:29 +0200 Subject: [PATCH] Update. --- problems.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/problems.py b/problems.py index 51e90ed..675a47a 100755 --- a/problems.py +++ b/problems.py @@ -308,7 +308,7 @@ class ProblemMixing(Problem): x = x.flatten(1) u = torch.arange(self.height * self.width).reshape(1, -1) m = ((x - u).abs() == 0).long() - d = (x - (m * u + (1-m) * self.height * self.width)).abs().sum(-1) + ( + d = (x - (m * u + (1 - m) * self.height * self.width)).abs().sum(-1) + ( m.sum(dim=-1) != self.height * self.width // 2 ).long() return d @@ -386,7 +386,7 @@ class ProblemMixing(Problem): #################### if __name__ == "__main__": - p = ProblemMixing(width=4, hard=True) + p = ProblemMixing() s, m = p.generate_sequences(10000) for x in s[:5]: print(p.seq2str(x)) -- 2.39.5