diff options
Diffstat (limited to 'Tanh.lua')
-rw-r--r-- | Tanh.lua | 5 |
1 files changed, 5 insertions, 0 deletions
@@ -4,3 +4,8 @@ function Tanh:__init(inplace) parent.__init(self, inplace) self.mode = 'CUDNN_ACTIVATION_TANH' end + +function Tanh:updateOutput(input) + if not self.mode then self.mode = 'CUDNN_ACTIVATION_TANH' end + return parent.updateOutput(self, input) +end |