Welcome to mirror list, hosted at ThFree Co, Russian Federation.

github.com/torch/nn.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
Diffstat (limited to 'Min.lua')
-rw-r--r--Min.lua16
1 files changed, 16 insertions, 0 deletions
diff --git a/Min.lua b/Min.lua
new file mode 100644
index 0000000..f4edbd8
--- /dev/null
+++ b/Min.lua
@@ -0,0 +1,16 @@
+local Min, parent = torch.class('nn.Min', 'nn.Module')
+
+function Min:__init(dimension)
+ parent.__init(self)
+ dimension = dimension or 1
+ self.dimension = dimension
+ self.indices = torch.Tensor()
+end
+
+function Min:updateOutput(input)
+ return input.nn.Min_updateOutput(self, input)
+end
+
+function Min:updateGradInput(input, gradOutput)
+ return input.nn.Min_updateGradInput(self, input, gradOutput)
+end