diff options
author | soumith <soumith@fb.com> | 2015-01-02 23:07:32 +0300 |
---|---|---|
committer | soumith <soumith@fb.com> | 2015-01-02 23:07:32 +0300 |
commit | df5925a2f094faf0d3cfd804cd7bea1617aab7df (patch) | |
tree | 07b47a5f8eca80230cdc713ee912c5e955de86bf /Tanh.lua | |
parent | e2595a3abfaea264ccbae70557ea3013961927be (diff) |
adding inplace non-linearities
Diffstat (limited to 'Tanh.lua')
-rw-r--r-- | Tanh.lua | 4 |
1 files changed, 2 insertions, 2 deletions
@@ -1,6 +1,6 @@ local Tanh, parent = torch.class('cudnn.Tanh','cudnn._Pointwise') -function Tanh:__init() - parent.__init(self) +function Tanh:__init(inplace) + parent.__init(self, inplace) self.mode = 'CUDNN_ACTIVATION_TANH' end |