nodes/mcDandy-more_math
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Forgot to reshape S in fft so it can be used

Browse Source
This commit is contained in:
mcDandy
2025-09-03 13:38:20 +02:00
parent bb579b21ad
commit 2799c6257e
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/more_math/Parser/TensorEvalVisitor.py
+1
View File
@@ -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())