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 /ReLU.lua | |
parent | 4b3f518fcbc21055f60bb7e18fa5b278d9cc8c79 (diff) |
fixes #1
Diffstat (limited to 'ReLU.lua')
-rw-r--r-- | ReLU.lua | 5 |
1 files changed, 3 insertions, 2 deletions
@@ -9,8 +9,9 @@ function ReLU:__init() end function ReLU: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) |