Class TransformerState

• Defined in File transformer.h

Inheritance Relationships

Base Type

• public marian::DecoderState (Class DecoderState)

Class Documentation

class TransformerState : public marian::DecoderState

Public Functions

TransformerState(const rnn::States &states, Logits logProbs, const std::vector<Ptr<EncoderState>> &encStates, Ptr<data::CorpusBatch> batch)

virtual Ptr<DecoderState> select(const std::vector<IndexType> &hypIndices, const std::vector<IndexType> &batchIndices, int beamSize) const