From 2799c6257e9a6e58da975fdc62f9d80d27095f4b Mon Sep 17 00:00:00 2001 From: mcDandy Date: Wed, 3 Sep 2025 13:38:20 +0200 Subject: [PATCH] Forgot to reshape S in fft so it can be used --- src/more_math/Parser/TensorEvalVisitor.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/more_math/Parser/TensorEvalVisitor.py b/src/more_math/Parser/TensorEvalVisitor.py index 76c1760..1e5faeb 100644 --- a/src/more_math/Parser/TensorEvalVisitor.py +++ b/src/more_math/Parser/TensorEvalVisitor.py @@ -145,6 +145,7 @@ class TensorEvalVisitor(MathExprVisitor): self.shape = time_shape shp_time = torch.zeros(time_shape, device=self.variables.get('device', 'cpu')) self.variables['T'] = getIndexTensorAlongDim(shp_time, 2) + self.variables['S'] = torch.full_like(shp_time, self.shape[2]) self.variables['B'] = getIndexTensorAlongDim(shp_time, 0) self.variables['C'] = getIndexTensorAlongDim(shp_time, 1) self.variables['R'] = torch.full_like(shp_time, self.variables['R'].flatten()[0].item())