TinyChatEngine
Loading...
Searching...
No Matches
Embedding Class Reference

Public Member Functions

 Embedding (int embed_dim_, int voc_size_, int padding_idx_, Matrix3D< float > lookup_)
 
void forward (Matrix3D< int > input_id, Matrix3D< float > output)
 

Public Attributes

int embed_dim
 
int voc_size
 
int padding_idx
 
Matrix3D< float > lookup
 

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