Allways stack when converting color representations

This commit is contained in:
Daniel Martinek
2026-03-13 15:26:14 +01:00
parent 7ce673175a
commit 96df6b6a53
+3 -10
View File
@@ -3284,11 +3284,7 @@ class UnifiedMathVisitor(MathExprVisitor):
# Value
v = max_rgb
# Return format
if num_args <= 2:
return torch.stack([h, s, v], dim=-1)
else:
return [h, s, v]
return torch.stack([h, s, v], dim=-1)
def visitHsvToRgbFunc(self, ctx):
num_args = len(ctx.expr())
@@ -3371,11 +3367,8 @@ class UnifiedMathVisitor(MathExprVisitor):
g = g + m
b = b + m
# Return format
if num_args <= 2:
return torch.stack([r, g, b], dim=-1)
else:
return [r, g, b]
return torch.stack([r, g, b], dim=-1)
def visitEntropyFunc(self, ctx):
val = self._promote_to_tensor((yield ctx.expr()))