[Feature] Import torch to Rspamd...

1 files changed, 19 insertions, 0 deletions

diff --git a/contrib/torch/nn/SoftMax.lua b/contrib/torch/nn/SoftMax.lua
new file mode 100644
index 000000000..23a444cf6
--- /dev/null
+++ b/contrib/torch/nn/SoftMax.lua
@@ -0,0 +1,19 @@
+local SoftMax, _ = torch.class('nn.SoftMax', 'nn.Module')
+
+function SoftMax:updateOutput(input)
+   input.THNN.SoftMax_updateOutput(
+      input:cdata(),
+      self.output:cdata()
+   )
+   return self.output
+end
+
+function SoftMax:updateGradInput(input, gradOutput)
+   input.THNN.SoftMax_updateGradInput(
+      input:cdata(),
+      gradOutput:cdata(),
+      self.gradInput:cdata(),
+      self.output:cdata()
+   )
+   return self.gradInput
+end