Merge pull request #161 from lvdmaaten/patch-2

Fix bicubic interpolation when source has size 1

1 files changed, 6 insertions, 0 deletions

diff --git a/generic/image.c b/generic/image.c
index 33a5293..0990814 100755
--- a/generic/image.c
+++ b/generic/image.c
@@ -166,6 +166,12 @@ static void image_(Main_scaleCubic_rowcol)(THTensor *Tsrc,
     long i;
     for( i = 0; i < dst_len; i++ )
       dst[ dst_start + i*dst_stride ] = src[ src_start + i*src_stride ];
+  } else if ( src_len == 1 ) {
+     long i;
+     for( i = 0; i < dst_len - 1; i++ ) {
+       long dst_pos = dst_start + i*dst_stride;
+       dst[dst_pos] = src[ src_start ];
+     }
   } else {
     long di;
     float si_f;