nmtSample::Projection Class Reference

calculates raw logits More...

Inheritance diagram for nmtSample::Projection:

Collaboration diagram for nmtSample::Projection:

Public Types
typedef std::shared_ptr< Projection >	ptr

Public Member Functions
	Projection ()=default
virtual void	addToModel (nvinfer1::INetworkDefinition *network, nvinfer1::ITensor *input, nvinfer1::ITensor **outputLogits)=0
	add raw logits to the network More...
virtual int	getOutputSize ()=0
	get the size of raw logits vector More...
	~Projection () override=default
virtual std::string	getInfo ()=0
	get the textual description of the component More...

Detailed Description

calculates raw logits

Member Typedef Documentation

ptr

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

Constructor & Destructor Documentation

Projection()

nmtSample::Projection::Projection ( )

default

~Projection()

nmtSample::Projection::~Projection ( )

overridedefault

Member Function Documentation

addToModel()

virtual void nmtSample::Projection::addToModel ( nvinfer1::INetworkDefinition *  network, nvinfer1::ITensor *  input, nvinfer1::ITensor **  outputLogits  )

pure virtual

add raw logits to the network

Implemented in nmtSample::SLPProjection.

getOutputSize()

virtual int nmtSample::Projection::getOutputSize ( )

pure virtual

get the size of raw logits vector

Implemented in nmtSample::SLPProjection.

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:

• projection.h