TinyChatEngine
|
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 |