Class SlicePooler

• Defined in File pooler.h

Inheritance Relationships

Base Type

• public marian::PoolerBase (Class PoolerBase)

Class Documentation

class SlicePooler : public marian::PoolerBase

Pool encoder state (contextual word embeddings) by selecting 1st embedding along sentence-length dimension.

Public Functions

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

std::vector<Expr> apply(Ptr<ExpressionGraph> graph, Ptr<data::CorpusBatch> batch, const std::vector<Ptr<EncoderState>> &encoderStates)

void clear()