Skip to main content

Module attn

zsfm_burn0.3.1

Module attn

Module Items

Functions

In crate zsfm_burn

Module attn

Functions§

burn_attention: Scaled dot-product attention with additive mask. q/k/v: [batch, heads, seq_q/kv, head_dim].