View source: R/attention-utils.R
query shape [batch, seqlen, filter_depth] key shape [batch, seqlen, filter_depth] value shape [batch, seqlen, filter_depth]
1 | create_qkv(x, filter_depth, num_parts = 1L, share_kv = FALSE)
|
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.