diff options
Diffstat (limited to 'src/rnn/rnn.h')
-rwxr-xr-x | src/rnn/rnn.h | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/rnn/rnn.h b/src/rnn/rnn.h index 74a535d3..c9131d87 100755 --- a/src/rnn/rnn.h +++ b/src/rnn/rnn.h @@ -75,11 +75,11 @@ private: std::vector<Expr> steps(xWs.size()); std::transform(xWs.begin(), xWs.end(), steps.begin(), [j](Expr e) { - return step(e, j, -3); + return slice(e, -3, j); }); if(mask) - state = cell_->applyState(steps, state, step(mask, j, -3)); + state = cell_->applyState(steps, state, slice(mask, -3, j)); else state = cell_->applyState(steps, state); |