diff --git a/src/deepali/core/kernels.py b/src/deepali/core/kernels.py index 0db123f..57857ff 100644 --- a/src/deepali/core/kernels.py +++ b/src/deepali/core/kernels.py @@ -282,7 +282,7 @@ def gaussian1d( else: if scale is None: scale = 1 - kernel = as_tensor(scale, dtype=dtype, device=device) + kernel = as_tensor([scale], dtype=dtype, device=device) return kernel