diff options
| author | Clement Farabet <clement.farabet@gmail.com> | 2014-07-15 01:28:21 +0400 |
|---|---|---|
| committer | Clement Farabet <clement.farabet@gmail.com> | 2014-07-15 01:28:21 +0400 |
| commit | 9e7733a04f9a1d3a42f9e9a2cd696b067f968651 (patch) | |
| tree | ab96ead4efa0c6d00d39d7cc3c5067c9890ae2eb /SpatialConvolutionMM.lua | |
| parent | 437c4940455ff8723a7318b2cb39a0970edde108 (diff) | |
preparing interface for CUDA
Diffstat (limited to 'SpatialConvolutionMM.lua')
| -rw-r--r-- | SpatialConvolutionMM.lua | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/SpatialConvolutionMM.lua b/SpatialConvolutionMM.lua index cc6a718..db908dd 100644 --- a/SpatialConvolutionMM.lua +++ b/SpatialConvolutionMM.lua @@ -1,13 +1,20 @@ local SpatialConvolutionMM, parent = torch.class('nn.SpatialConvolutionMM', 'nn.Module') -function SpatialConvolutionMM:__init(nInputPlane, nOutputPlane, kW, kH) +function SpatialConvolutionMM:__init(nInputPlane, nOutputPlane, kW, kH, dW, dH, padding) parent.__init(self) + + dW = dW or 1 + dH = dH or 1 self.nInputPlane = nInputPlane self.nOutputPlane = nOutputPlane self.kW = kW self.kH = kH + self.dW = dW + self.dH = dH + self.padding = padding or 0 + self.weight = torch.Tensor(nOutputPlane, nInputPlane*kH*kW) self.bias = torch.Tensor(nOutputPlane) self.gradWeight = torch.Tensor(nOutputPlane, nInputPlane*kH*kW) |