diff options
author | Clement Farabet <clement.farabet@gmail.com> | 2011-11-11 19:38:11 +0400 |
---|---|---|
committer | Clement Farabet <clement.farabet@gmail.com> | 2011-11-11 19:38:11 +0400 |
commit | 98fb40a126afb2f33800af2f081d3119a8b55448 (patch) | |
tree | b811fd62ac82766bc9d1a49f4d9e99cdc52ec8fa /SpatialUpSampling.lua | |
parent | a8ab27537e16f4440565fbebe6313b2172b489a1 (diff) |
Upgraded all code to new torch master.torch-master
Diffstat (limited to 'SpatialUpSampling.lua')
-rw-r--r-- | SpatialUpSampling.lua | 20 |
1 files changed, 4 insertions, 16 deletions
diff --git a/SpatialUpSampling.lua b/SpatialUpSampling.lua index 90d0126..0d26bcd 100644 --- a/SpatialUpSampling.lua +++ b/SpatialUpSampling.lua @@ -25,26 +25,14 @@ function SpatialUpSampling:__init(...) {arg='dH', type='number', help='stride height', req=true}) end -function SpatialUpSampling:forward(input) +function SpatialUpSampling:updateOutput(input) self.output:resize(input:size(1), input:size(2) * self.dH, input:size(3) * self.dW) - input.nn.SpatialUpSampling_forward(self, input) + input.nn.SpatialUpSampling_updateOutput(self, input) return self.output end -function SpatialUpSampling:backward(input, gradOutput) +function SpatialUpSampling:updateGradInput(input, gradOutput) self.gradInput:resizeAs(input) - input.nn.SpatialUpSampling_backward(self, input, gradOutput) + input.nn.SpatialUpSampling_updateGradInput(self, input, gradOutput) return self.gradInput end - -function SpatialUpSampling:write(file) - parent.write(self, file) - file:writeInt(self.dW) - file:writeInt(self.dH) -end - -function SpatialUpSampling:read(file) - parent.read(self, file) - self.dW = file:readInt() - self.dH = file:readInt() -end |