Expand description
ยงStochastic Attention
Rate-mode and SC-mode attention primitives used by the Python bridge.
Structsยง
Functionsยง
- extract_
head_ ๐columns - Extract one head slice from a row-major matrix.
- flatten_
rows ๐ - validate_
multihead_ ๐shapes - validate_
shapes ๐