diff options
Diffstat (limited to 'SpatialDownSampling.lua')
-rw-r--r-- | SpatialDownSampling.lua | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/SpatialDownSampling.lua b/SpatialDownSampling.lua index b18849f..2aa4216 100644 --- a/SpatialDownSampling.lua +++ b/SpatialDownSampling.lua @@ -26,6 +26,11 @@ function SpatialDownSampling:__init(...) end function SpatialDownSampling:updateOutput(input) + if (input:size(2) / self.rH) < 1 then + error('input too small in dimension 2') + elseif (input:size(3) / self.rW) < 1 then + error('input too small in dimension 3') + end self.output:resize(input:size(1), math.floor(input:size(2) / self.rH), math.floor(input:size(3) / self.rW)) input.nn.SpatialDownSampling_updateOutput(self, input) |