dl::Embedding Class Reference

Inheritance diagram for dl::Embedding:

Collaboration diagram for dl::Embedding:

Public Member Functions
	Embedding (size_t numEmbeddings, size_t embeddingDim)
virtual dl::TensorPtr	forward (const dl::TensorPtr input)

Detailed Description

Definition at line 8 of file embedding.hpp.

Constructor & Destructor Documentation

Embedding()

dl::Embedding::Embedding ( size_t  numEmbeddings, size_t  embeddingDim  )

inline

Definition at line 13 of file embedding.hpp.

                                                             : weight(dl::empty({numEmbeddings, embeddingDim})) {
            registerParameter("weight", weight);
        }

Member Function Documentation

forward()

virtual dl::TensorPtr dl::Embedding::forward ( const dl::TensorPtr  input )

inlinevirtual

Todo:

implement

Definition at line 18 of file embedding.hpp.

                                                             {
            throw std::runtime_error("Not yet implemented");
        }

---

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

• dl/model/embedding.hpp