diff options
author | Sergey Zagoruyko <zagoruyko2@gmail.com> | 2016-02-26 01:51:16 +0300 |
---|---|---|
committer | Sergey Zagoruyko <zagoruyko2@gmail.com> | 2016-02-26 01:51:16 +0300 |
commit | 1f8258ac5566c47a06d12710f71cd15d728de8b3 (patch) | |
tree | 84a6976a58c37549b0e98dde5c2818e5fde9cbd0 /VolumetricAveragePooling.lua | |
parent | b0d9efc1d7a3dfa7eda0a9bafa15d75ebb6d15c0 (diff) |
__tostring__ for Volumetric modules
Diffstat (limited to 'VolumetricAveragePooling.lua')
-rw-r--r-- | VolumetricAveragePooling.lua | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/VolumetricAveragePooling.lua b/VolumetricAveragePooling.lua index f5adcd0..41cd46f 100644 --- a/VolumetricAveragePooling.lua +++ b/VolumetricAveragePooling.lua @@ -40,3 +40,15 @@ end function VolumetricAveragePooling:empty() return parent.clearState(self) end + +function VolumetricAveragePooling:__tostring__() + local s = string.format('%s(%d,%d,%d,%d,%d,%d', torch.type(self), + self.kT, self.kW, self.kH, self.dT, self.dW, self.dH) + if (self.padT or self.padW or self.padH) and + (self.padT ~= 0 or self.padW ~= 0 or self.padH ~= 0) then + s = s .. ',' .. self.padT.. ',' .. self.padW .. ','.. self.padH + end + s = s .. ')' + + return s +end |