nmtSample::Embedder Class Reference

projects 1-hot vectors (represented as a vector with indices) into dense embedding space More...

Inheritance diagram for nmtSample::Embedder:

Collaboration diagram for nmtSample::Embedder:

Public Types
typedef std::shared_ptr< Embedder >	ptr

Public Member Functions
	Embedder ()=default
virtual void	addToModel (nvinfer1::INetworkDefinition *network, nvinfer1::ITensor *input, nvinfer1::ITensor **output)=0
	add the embedding vector calculation to the network More...
virtual int	getInputDimensionSize ()=0
	get the upper bound for the possible values of indices More...
	~Embedder () override=default
virtual std::string	getInfo ()=0
	get the textual description of the component More...

Detailed Description

projects 1-hot vectors (represented as a vector with indices) into dense embedding space

Member Typedef Documentation

ptr

typedef std::shared_ptr<Embedder> nmtSample::Embedder::ptr

Constructor & Destructor Documentation

Embedder()

nmtSample::Embedder::Embedder ( )

default

~Embedder()

nmtSample::Embedder::~Embedder ( )

overridedefault

Member Function Documentation

addToModel()

virtual void nmtSample::Embedder::addToModel ( nvinfer1::INetworkDefinition *  network, nvinfer1::ITensor *  input, nvinfer1::ITensor **  output  )

pure virtual

add the embedding vector calculation to the network

Implemented in nmtSample::SLPEmbedder.

getInputDimensionSize()

virtual int nmtSample::Embedder::getInputDimensionSize ( )

pure virtual

get the upper bound for the possible values of indices

Implemented in nmtSample::SLPEmbedder.

getInfo()

virtual std::string nmtSample::Component::getInfo ( )

pure virtualinherited

get the textual description of the component

Implemented in nmtSample::SoftmaxLikelihood, nmtSample::BeamSearchPolicy, nmtSample::MultiplicativeAlignment, nmtSample::SLPEmbedder, nmtSample::SLPProjection, nmtSample::BLEUScoreWriter, nmtSample::Context, nmtSample::LSTMEncoder, nmtSample::TextWriter, nmtSample::SLPAttention, nmtSample::BenchmarkWriter, nmtSample::TextReader, nmtSample::LSTMDecoder, and nmtSample::LimitedSamplesDataReader.

---

The documentation for this class was generated from the following file:

• embedder.h