diff options
| author | Alfredo Canziani <alfredo.canziani@gmail.com> | 2016-03-02 06:12:12 +0300 |
|---|---|---|
| committer | Alfredo Canziani <alfredo.canziani@gmail.com> | 2016-03-02 06:12:12 +0300 |
| commit | fadfadb961e492f0da22e95386e96c1bffff019c (patch) | |
| tree | abb3921d29fc15b91b7a9f9d8a3192db5ba761cb /SpatialZeroPadding.lua | |
| parent | 260cc314fe5336b1d8bff56294a135a052514d7c (diff) | |
Better __tostring__ and cleans formatting
Diffstat (limited to 'SpatialZeroPadding.lua')
| -rw-r--r-- | SpatialZeroPadding.lua | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/SpatialZeroPadding.lua b/SpatialZeroPadding.lua index 72d1c63..f199258 100644 --- a/SpatialZeroPadding.lua +++ b/SpatialZeroPadding.lua @@ -58,7 +58,7 @@ function SpatialZeroPadding:updateOutput(input) end function SpatialZeroPadding:updateGradInput(input, gradOutput) - if input:dim() == 3 then + if input:dim() == 3 then self.gradInput:resizeAs(input):zero() -- crop gradInput if necessary local cg_input = self.gradInput @@ -98,7 +98,7 @@ end function SpatialZeroPadding:__tostring__() - return torch.type(self) .. - string.format('(l=%d,r=%d,t=%d,b=%d)', self.pad_l, self.pad_r, - self.pad_t, self.pad_b) + return torch.type(self) .. + string.format('(l=%d, r=%d, t=%d, b=%d)', self.pad_l, self.pad_r, + self.pad_t, self.pad_b) end |