From 58287392fbee12cd5da1f58c066e04aa55734663 Mon Sep 17 00:00:00 2001 From: Daniel Martinek Date: Sun, 5 Apr 2026 13:00:50 +0200 Subject: [PATCH] fix logspace not using base --- more_math/Parser/UnifiedMathVisitor.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/more_math/Parser/UnifiedMathVisitor.py b/more_math/Parser/UnifiedMathVisitor.py index 268e710..56d1b29 100644 --- a/more_math/Parser/UnifiedMathVisitor.py +++ b/more_math/Parser/UnifiedMathVisitor.py @@ -3498,14 +3498,14 @@ class UnifiedMathVisitor(MathExprVisitor): start_val = yield ctx.expr(0) end_val = yield ctx.expr(1) steps_val = yield ctx.expr(2) - base_val = yield ctx.expr(2) + base_val = yield ctx.expr(3) start = float(start_val.item()) if self._is_tensor(start_val) else float(start_val) end = float(end_val.item()) if self._is_tensor(end_val) else float(end_val) base = float(base_val.item()) if self._is_tensor(base_val) else float(base_val) steps = int(steps_val.item()) if self._is_tensor(steps_val) else int(steps_val) - return torch.logspace(start, end, steps, device=self.device) + return torch.logspace(start, end, steps, base=base, device=self.device) def visitRollFunc(self, ctx): """roll(x, shift, [dim]) - circular shift of elements"""