std1.0.0-beta-7 SafeC docs

std :: ml::attention

Function std::mha_forward_mask

#include <std/ml/attention.h> source ยท definition

&FloatTensor mha_forward_mask(
    const &FloatTensor Q,
    const &FloatTensor K,
    const &FloatTensor V,
    unsigned long numHeads,
    int causal,
    unsigned long window,
)