Fix for using non-torch.DoubleTensor inputs to image.gaussianpyramid

1 files changed, 3 insertions, 3 deletions

diff --git a/init.lua b/init.lua
index 89aa3a7..b19abf9 100755
--- a/init.lua
+++ b/init.lua
@@ -1954,18 +1954,18 @@ function image.gaussianpyramid(...)
    end
    if src:nDimension() == 2 then
       for i = 1,#scales do
-         dst[i] = dst[i] or torch.Tensor()
+         dst[i] = dst[i] or src.new()
          dst[i]:resize(src:size(1)*scales[i], src:size(2)*scales[i])
       end
    elseif src:nDimension() == 3 then
       for i = 1,#scales do
-         dst[i] = dst[i] or torch.Tensor()
+         dst[i] = dst[i] or src.new()
          dst[i]:resize(src:size(1), src:size(2)*scales[i], src:size(3)*scales[i])
       end
    else
       dok.error('src image must be 2D or 3D', 'image.gaussianpyramid')
    end
-   local k = image.gaussian{width=3, normalize=true}
+   local k = image.gaussian{width=3, normalize=true}:typeAs(src)
    local tmp = src
    for i = 1,#scales do
       if scales[i] == 1 then