diff options
| author | Michael 'myrhev' Mathieu <michael.mathieu@ens.fr> | 2012-06-14 22:34:59 +0400 |
|---|---|---|
| committer | Michael 'myrhev' Mathieu <michael.mathieu@ens.fr> | 2012-06-14 22:34:59 +0400 |
| commit | bee7a2b96c80caafb78130f627097372489b7503 (patch) | |
| tree | e159bf1c55f8496f0730641789720fd4f3ab54f8 /SpatialPadding.lua | |
| parent | 07562f3244cb7c85f71fb47be7d57a2a66c7fd98 (diff) | |
Add SpatialRadialMatching
Diffstat (limited to 'SpatialPadding.lua')
| -rw-r--r-- | SpatialPadding.lua | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/SpatialPadding.lua b/SpatialPadding.lua index c542c16..d8a1951 100644 --- a/SpatialPadding.lua +++ b/SpatialPadding.lua @@ -24,6 +24,9 @@ function SpatialPadding:__init(pad_l, pad_r, pad_t, pad_b, y_dim, x_dim) end function SpatialPadding:updateOutput(input) + if self.output:type() ~= input:type() then + self.output = input.new() + end self.x_dim = self.x_dim or 3 self.y_dim = self.y_dim or 2 local h = input:size(self.y_dim) + self.pad_t + self.pad_b |