Fixed conv filter channel dimension.

mikgroup · Aug 23, 2018 · 8e0986d · 8e0986d
1 parent 4731637
commit 8e0986d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sigpy/linop.py b/sigpy/linop.py
@@ -1515,10 +1515,10 @@ def __init__(self, y_shape, x, ndim, mode='full',
             W_shape = [m - p + 1 for m, p in zip(x.shape[-ndim:], y_shape[-ndim:])]
 
         if input_multi_channel:
-            W_shape = [x.shape[0]] + W_shape
+            W_shape = [x.shape[-ndim - 1]] + W_shape
 
         if output_multi_channel:
-            W_shape = [y_shape[0]] + W_shape
+            W_shape = [y_shape[-ndim - 1]] + W_shape
 
         super().__init__(W_shape, y_shape)