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

Public Member Functions

 RotaryPosEmb (Matrix3D< float > _cos, Matrix3D< float > _sin, std::string path)
 
void forward (Matrix3D< float > &key, Matrix3D< float > &value, int start_idx, int len)
 

Public Attributes

Matrix3D< float > cos
 
Matrix3D< float > sin
 

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