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