diff options
author | Sergey Zagoruyko <zagoruyko2@gmail.com> | 2016-02-04 17:53:04 +0300 |
---|---|---|
committer | Sergey Zagoruyko <zagoruyko2@gmail.com> | 2016-02-09 21:08:41 +0300 |
commit | b4ebdf2f95ee9f1429825a0d7b0948721e407d82 (patch) | |
tree | 77b6183d0c258d07d5f5e6338c48ccb5fc5a877b /Reshape.lua | |
parent | 4a433463fcd17c4ec3242586da7e9fc859feb3ae (diff) |
nn.clearState
Diffstat (limited to 'Reshape.lua')
-rw-r--r-- | Reshape.lua | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/Reshape.lua b/Reshape.lua index dcc787b..80f8b01 100644 --- a/Reshape.lua +++ b/Reshape.lua @@ -67,3 +67,8 @@ function Reshape:__tostring__() return torch.type(self) .. '(' .. table.concat(self.size:totable(), 'x') .. ')' end + +function Reshape:clearState() + nn.utils.clear(self, '_input', '_gradOutput') + return parent.clearState(self) +end |