multi_head: Multi-Headed Attention

View source: R/multi_head.R

multi_headR Documentation

Multi-Headed Attention

Description

Multi-Headed Attention

Usage

multi_head(Q, K, V, d_model, num_heads, mask = NULL)

Arguments

Q

queries

K

keys

V

values

d_model

dimensions of the model

num_heads

number of heads

mask

optional mask

Value

multi-headed attention outputs


transformer documentation built on Nov. 10, 2023, 5:08 p.m.