projects
/
culture.git
/ commitdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
| commitdiff |
tree
raw
|
patch
|
inline
| side by side (parent:
e08d6e7
)
Update.
author
François Fleuret
<francois@fleuret.org>
Sat, 6 Jul 2024 05:28:12 +0000
(08:28 +0300)
committer
François Fleuret
<francois@fleuret.org>
Sat, 6 Jul 2024 05:28:12 +0000
(08:28 +0300)
reasoning.py
patch
|
blob
|
history
diff --git
a/reasoning.py
b/reasoning.py
index
951e04a
..
fb208b0
100755
(executable)
--- a/
reasoning.py
+++ b/
reasoning.py
@@
-117,7
+117,7
@@
class Reasoning(problem.Problem):
c = c.long()[:, None]
c = (
(1 - ((c == 1).long() + (c == 0).long() + (c == -1).long()))
c = c.long()[:, None]
c = (
(1 - ((c == 1).long() + (c == 0).long() + (c == -1).long()))
- * torch.tensor([
128, 128, 128
], device=c.device)
+ * torch.tensor([
64, 64, 64
], device=c.device)
+ (c == 1).long() * torch.tensor([0, 255, 0], device=c.device)
+ (c == 0).long() * torch.tensor([255, 255, 255], device=c.device)
+ (c == -1).long() * torch.tensor([255, 0, 0], device=c.device)
+ (c == 1).long() * torch.tensor([0, 255, 0], device=c.device)
+ (c == 0).long() * torch.tensor([255, 255, 255], device=c.device)
+ (c == -1).long() * torch.tensor([255, 0, 0], device=c.device)