Skip to content

Commit

Permalink
[core] Gaussian kernel tensor ndim must be 1 even when radius=0
Browse files Browse the repository at this point in the history
  • Loading branch information
aschuh-hf committed Oct 30, 2023
1 parent b2a9322 commit 1fdab64
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/deepali/core/kernels.py
Original file line number Diff line number Diff line change
Expand Up @@ -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


Expand Down

0 comments on commit 1fdab64

Please sign in to comment.