Fixed triangular window for multidimensional.

mikgroup · Jul 22, 2018 · 09ed7c1 · 09ed7c1
1 parent b262c97
commit 09ed7c1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sigpy/util.py b/sigpy/util.py
@@ -602,7 +602,7 @@ def triang(shape, dtype=np.complex, device=cpu_device):
     xp = device.xp
 
     with device:
-        window = 1
+        window = ones(shape, dtype=dtype, device=device)
         for n, i in enumerate(shape[::-1]):
             w = 1 - xp.abs(xp.arange(i, dtype=dtype) - i // 2 + ((i + 1) % 2) / 2) / ((i + 1) // 2)
             window *= w.reshape([i] + [1] * n)