Class MaxPooler

• Defined in File pooler.h

Inheritance Relationships

Base Type

• public marian::PoolerBase (Class PoolerBase)

Class Documentation

class MaxPooler : public marian::PoolerBase

Pool encoder state (contextual word embeddings) via max-pooling along sentence-length dimension.

Public Functions

MaxPooler(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()