create_qkv: Takes input tensor of shape [batch, seqlen, channels] and...

Description Usage

View source: R/attention-utils.R

Description

query shape [batch, seqlen, filter_depth] key shape [batch, seqlen, filter_depth] value shape [batch, seqlen, filter_depth]

Usage

1
create_qkv(x, filter_depth, num_parts = 1L, share_kv = FALSE)

ifrit98/transformR documentation built on Nov. 26, 2019, 2:14 a.m.