diff options
author | Soumith Chintala <soumith@gmail.com> | 2014-09-28 23:31:53 +0400 |
---|---|---|
committer | Soumith Chintala <soumith@gmail.com> | 2014-09-28 23:55:38 +0400 |
commit | 91793f499ab03a2d89cfa0148711e4ce02ab31ba (patch) | |
tree | 311aeb793a448dc87b67f6e1dd1d95ec75b4305a /SoftMax.lua | |
parent | 2f8a13181e33e05261077c8108e39edc18bf0f3e (diff) |
adding SoftMax and SpatialSoftMax bindings
Diffstat (limited to 'SoftMax.lua')
-rw-r--r-- | SoftMax.lua | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/SoftMax.lua b/SoftMax.lua new file mode 100644 index 0000000..41bffaa --- /dev/null +++ b/SoftMax.lua @@ -0,0 +1,9 @@ +local SoftMax, parent = torch.class('cudnn.SoftMax', 'cudnn.SpatialSoftMax') +local ffi = require 'ffi' +local C = cudnn.C +local errcheck = cudnn.errcheck + +function SoftMax:__init(fast) + parent.__init(self, fast) + self.mode = 'CUDNN_SOFTMAX_MODE_INSTANCE' +end |