Class EncoderDecoderLayerBase

• Defined in File generic.h

Inheritance Relationships

Base Type

• public marian::LayerBase (Class LayerBase)

Derived Types

• public marian::DecoderBase (Class DecoderBase)

• public marian::EncoderBase (Class EncoderBase)

Class Documentation

class EncoderDecoderLayerBase : public marian::LayerBase

Base class for Encoder and Decoder classes.

Have embeddings and a batch index (=stream index).

Subclassed by marian::DecoderBase, marian::EncoderBase

Public Functions

Ptr<IEmbeddingLayer> getEmbeddingLayer(bool ulr = false) const

Get all embedding layer(s).

It lazily creates the embedding layer on first call. This is lazy mostly because the constructors of the consuming objects are not guaranteed presently to have access to their graph.

Return

a shared pointer to the embedding layer

Parameters

• ulr: whether to use ULREmbedding layer. false by default.

Protected Functions

EncoderDecoderLayerBase(Ptr<ExpressionGraph> graph, Ptr<Options> options, const std::string &prefix, size_t batchIndex, float dropoutEmbeddings, bool embeddingFix)

virtual ~EncoderDecoderLayerBase()

Protected Attributes

const std::string prefix_

const bool embeddingFix_

const float dropoutEmbeddings_

const bool inference_

const size_t batchIndex_

std::vector<Ptr<IEmbeddingLayer>> embeddingLayers_