Merge pull request #63 from jonathantompson/hflip_vflip_assert_fix

modifed h/vflip assert to only check if 2D tensors are contiguous (maintaining backward compat.)

1 files changed, 4 insertions, 4 deletions

diff --git a/init.lua b/init.lua
index 5fc8243..d204d1e 100755
--- a/init.lua
+++ b/init.lua
@@ -813,8 +813,8 @@ local function hflip(...)
       dok.error('incorrect arguments', 'image.hflip')
    end
 
-   if not src:isContiguous() then
-     dok.error('input tensor is not contiguous', 'image.hflip')
+   if (src:dim() == 2) and (not src:isContiguous()) then
+     dok.error('2D input tensor is not contiguous', 'image.hflip')
    end
 
    dst = dst or src.new()
@@ -855,8 +855,8 @@ local function vflip(...)
       dok.error('incorrect arguments', 'image.vflip')
    end
 
-   if not src:isContiguous() then
-     dok.error('input tensor is not contiguous', 'image.vflip')
+   if (src:dim() == 2) and (not src:isContiguous()) then
+     dok.error('2D input tensor is not contiguous', 'image.vflip')
    end
 
    dst = dst or src.new()