diff options
| author | Soumith Chintala <soumith@gmail.com> | 2014-09-11 01:54:08 +0400 |
|---|---|---|
| committer | Soumith Chintala <soumith@gmail.com> | 2014-09-21 03:51:04 +0400 |
| commit | ecd1ca60adb49dab5eaa317dbb17bf80c540c8ef (patch) | |
| tree | 0f08e1be714e7418096f5b2b78d4c41929361128 /Tanh.lua | |
| parent | 4b3f518fcbc21055f60bb7e18fa5b278d9cc8c79 (diff) | |
fixes #1
Diffstat (limited to 'Tanh.lua')
| -rw-r--r-- | Tanh.lua | 5 |
1 files changed, 3 insertions, 2 deletions
@@ -9,8 +9,9 @@ function Tanh:__init() end function Tanh:createIODescriptors(input) - if input:size(1) ~= self.iSize:size(1) or input:size(2) ~= self.iSize:size(2) - or input:size(3) ~= self.iSize:size(3) or input:size(4) ~= self.iSize:size(4) then + if input:size(1) ~= self.iSize[1] or input:size(2) ~= self.iSize[2] + or input:size(3) ~= self.iSize[3] or input:size(4) ~= self.iSize[4] then + self.iSize = input:size() self.gradInput:resizeAs(input) self.output:resizeAs(input) self.iDesc = cudnn.toDescriptor(input) |