API ReferenceAPI Reference·ModulesModules·MlMloa::ByteHeadBYTE OUTPUT HEAD - convert hidden states back to byte probabilities input: [batch, seq_len, d_model] output: [batch, seq_len, 256] (logits over byte values)ClC++ public headers#include <oa/ml/byte.h>#include <oa/ml/byte.h>Inheritancepublic ModuleOpen class hierarchyOpen class hierarchyPublic MethodsforwardforwardMatrix oa::ByteHead::forward(const Matrix & inHidden)Constructor & Destructor Documentationoa::ByteHead::ByteHead( oa::I32 inDModel )SourceSourceNo public source comment is attached to this declaration.ParametersinDModeloa::I32—Public Method DocumentationMatrix oa::ByteHead::forward( const Matrix & inHidden )SourceSource[batch, seq, d_model] -> [batch, seq, 256]ParametersinHiddenconst Matrix &—ReturnsMatrixThe declared return value.
oa::ByteHead::ByteHead( oa::I32 inDModel )SourceSourceNo public source comment is attached to this declaration.ParametersinDModeloa::I32—
Matrix oa::ByteHead::forward( const Matrix & inHidden )SourceSource[batch, seq, d_model] -> [batch, seq, 256]ParametersinHiddenconst Matrix &—ReturnsMatrixThe declared return value.
