layer_compute_qkv_v2: Split input into query, key, value matrices in preparation...

Description Usage

View source: R/attention-layers.R

Description

Done a little differently than T2T framework. TODO: (TEST THIS!)

Usage

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

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