From 223dcead67d6d0236dfef37ddd655110f1122c58 Mon Sep 17 00:00:00 2001
From: nagadomi <nagadomi@nurs.or.jp>
Date: Sat, 12 Mar 2016 06:54:09 +0900
Subject: [PATCH] Add support for clearState() in w2nn.LeakyReLU

---
 lib/LeakyReLU.lua | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/lib/LeakyReLU.lua b/lib/LeakyReLU.lua
index 5b27bc9..35a0cde 100644
--- a/lib/LeakyReLU.lua
+++ b/lib/LeakyReLU.lua
@@ -17,7 +17,7 @@ function LeakyReLU:updateOutput(input)
    
    return self.output
 end
- 
+
 function LeakyReLU:updateGradInput(input, gradOutput)
    self.gradInput:resizeAs(gradOutput)
    -- filter positive
@@ -29,3 +29,8 @@ function LeakyReLU:updateGradInput(input, gradOutput)
    
    return self.gradInput
 end
+
+function LeakyReLU:clearState()
+   nn.utils.clear(self, 'negative')
+   return parent.clearState(self)
+end