diff options
Diffstat (limited to 'src/models/states.h')
-rw-r--r-- | src/models/states.h | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/src/models/states.h b/src/models/states.h index f1f3c704..e2635c87 100644 --- a/src/models/states.h +++ b/src/models/states.h @@ -48,8 +48,10 @@ public: virtual Expr getProbs() { return probs_; } virtual void setProbs(Expr probs) { probs_ = probs; } - virtual Ptr<DecoderState> select(const std::vector<size_t>& selIdx, int beamSize) { - return New<DecoderState>(states_.select(selIdx, beamSize), probs_, encStates_); + virtual Ptr<DecoderState> select(const std::vector<size_t>& selIdx, + int beamSize) { + return New<DecoderState>( + states_.select(selIdx, beamSize), probs_, encStates_); } virtual const rnn::States& getStates() { return states_; } |