Class DecoderBase

• Defined in File decoder.h

Inheritance Relationships

Base Type

• public marian::EncoderDecoderLayerBase (Class EncoderDecoderLayerBase)

Derived Types

• public marian::DecoderS2S (Class DecoderS2S)

• public marian::Transformer< DecoderBase > (Template Class Transformer)

Class Documentation

class DecoderBase : public marian::EncoderDecoderLayerBase

Subclassed by marian::DecoderS2S, marian::Transformer< DecoderBase >

Public Functions

DecoderBase(Ptr<ExpressionGraph> graph, Ptr<Options> options)

virtual Ptr<DecoderState> startState(Ptr<ExpressionGraph>, Ptr<data::CorpusBatch> batch, std::vector<Ptr<EncoderState>>&) = 0

virtual Ptr<DecoderState> step(Ptr<ExpressionGraph>, Ptr<DecoderState>) = 0

virtual void embeddingsFromBatch(Ptr<ExpressionGraph> graph, Ptr<DecoderState> state, Ptr<data::CorpusBatch> batch)

virtual void embeddingsFromPrediction(Ptr<ExpressionGraph> graph, Ptr<DecoderState> state, const Words &words, int dimBatch, int dimBeam)

virtual const std::vector<Expr> getAlignments(int = 0)

virtual Ptr<data::Shortlist> getShortlist()

virtual void setShortlist(Ptr<data::Shortlist> shortlist)

virtual void clear() = 0

Protected Attributes

Ptr<data::Shortlist> shortlist_