diff options
| author | Gregory Chanan <gchanan@fb.com> | 2016-11-15 23:50:08 +0300 |
|---|---|---|
| committer | Gregory Chanan <gchanan@fb.com> | 2016-11-15 23:50:08 +0300 |
| commit | 1828978f428a2df6409e8e0e7851e5b932a07c0c (patch) | |
| tree | 825372aa5584782f99f2c25fb4c40b22d455fdc7 /SpatialUpSamplingBilinear.lua | |
| parent | 53e73cc687195ab42fe03ed0947a76cb515cddad (diff) | |
Add contiguous checking / make tensors contiguous for
SpatialUpSamplingBilinear, PReLU, SpatialSubSampling, TemporalConvolution.
Diffstat (limited to 'SpatialUpSamplingBilinear.lua')
| -rw-r--r-- | SpatialUpSamplingBilinear.lua | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/SpatialUpSamplingBilinear.lua b/SpatialUpSamplingBilinear.lua index 54ce5b8..8f19f91 100644 --- a/SpatialUpSamplingBilinear.lua +++ b/SpatialUpSamplingBilinear.lua @@ -70,12 +70,12 @@ end function SpatialUpSamplingBilinear:updateOutput(input) assert(input:dim() == 4 or input:dim()==3, 'SpatialUpSamplingBilinear only supports 3D or 4D tensors' ) + input = makeContiguous(self, input) local inputwas3D = false if input:dim() == 3 then input=input:view(-1, input:size(1), input:size(2), input:size(3)) inputwas3D = true end - input = makeContiguous(self, input) local xdim = input:dim() local ydim = xdim - 1 self:setSize(input) @@ -98,6 +98,7 @@ function SpatialUpSamplingBilinear:updateGradInput(input, gradOutput) 'SpatialUpSamplingBilinear only support 3D or 4D tensors' ) assert(input:dim() == gradOutput:dim(), 'Input and gradOutput should be of same dimension' ) + input, gradOutput = makeContiguous(self, input, gradOutput) local inputwas3D = false if input:dim() == 3 then input = input:view(-1, input:size(1), input:size(2), input:size(3)) |