diff options
author | Sasank Chilamkurthy <sasankchilamkurthy@gmail.com> | 2016-08-11 15:11:23 +0300 |
---|---|---|
committer | Sasank Chilamkurthy <sasankchilamkurthy@gmail.com> | 2016-08-11 15:11:23 +0300 |
commit | a2e5587fa931a7d0fbda4194354e200763656e85 (patch) | |
tree | 43d95612409a4b4c1c8c8f37d678c50d91ac5b49 /VolumetricLogSoftMax.lua | |
parent | 59ac9afc330e3b848fb204afeb8d6f2835902540 (diff) |
Add cudnn.VolumetricLogSoftMax cudnn.VolumetricSoftMax.lua
Diffstat (limited to 'VolumetricLogSoftMax.lua')
-rw-r--r-- | VolumetricLogSoftMax.lua | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/VolumetricLogSoftMax.lua b/VolumetricLogSoftMax.lua new file mode 100644 index 0000000..a23ed60 --- /dev/null +++ b/VolumetricLogSoftMax.lua @@ -0,0 +1,7 @@ +local SoftMax, parent = torch.class('cudnn.VolumetricLogSoftMax', 'cudnn.VolumetricSoftMax') + +function SoftMax:__init(fast) + parent.__init(self, fast) + self.ssm.mode = 'CUDNN_SOFTMAX_MODE_CHANNEL' + self.ssm.algorithm = 'CUDNN_SOFTMAX_LOG' +end |