split_heads: Split channels (dimension 2) into multiple heads (becomes...
In ifrit98/transformR: An R Implementation of Transformer Model with Self Attention
View source: R/attention-utils.R
Description
Split channels (dimension 2) into multiple heads (becomes dimension 1).
x Tensor shape: [batch, length, channels]
num_heads integer
Usage
1
split_heads(x, num_heads)
ifrit98/transformR documentation built on Nov. 26, 2019, 2:14 a.m.
R Package Documentation
Browse R Packages
We want your feedback!
Note that we can't provide technical support on individual packages. You should contact the package authors for that.
View source: R/attention-utils.R
Description
Split channels (dimension 2) into multiple heads (becomes dimension 1). x Tensor shape: [batch, length, channels] num_heads integer
Usage
1 | split_heads(x, num_heads)
|
R Package Documentation
Browse R Packages
We want your feedback!
Note that we can't provide technical support on individual packages. You should contact the package authors for that.
Embedding an R snippet on your website
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.