Hacker News new | ask | show | jobs
by ctrw 778 days ago
You mean like

    res2 = dumbsum(query, key, 'batch seq_q d_model, batch seq_k d_model -> batch seq_q seq_k')