layer_self_attention_simple: Simplified Self attention layer Expecting shape(x) == (batch,...

Description Usage

View source: R/attention-layers.R

Description

Simplified Self attention layer Expecting shape(x) == (batch, maxtime, units)

Usage

1
2
3
4
5
6
7
8
layer_self_attention_simple(
  x,
  filter_depth = 32L,
  output_depth = 64L,
  num_parts = 3L,
  dropout = 0,
  share_kv = TRUE
)

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