Merge pull request #6 from yimmon/master

support batch_resize (sequence training related layers are in kaldi_seq)

1 files changed, 6 insertions, 0 deletions

diff --git a/nerv/layer/combiner.lua b/nerv/layer/combiner.lua
index 7bd7617..1bcfdfb 100644
--- a/nerv/layer/combiner.lua
+++ b/nerv/layer/combiner.lua
@@ -30,6 +30,12 @@ function CombinerLayer:init(batch_size)
     self.sum = self.gconf.cumat_type(batch_size, dim)
 end
 
+function CombinerLayer:batch_resize(batch_size)
+    if self.sum:nrow() ~= batch_size then
+        self.sum = self.gconf.cumat_type(batch_size, self.dim_in[1])
+    end
+end
+
 function CombinerLayer:update(bp_err, input, output)
 end