Packages

viva_tensor

2.2.103
2.2.104 2.2.103 2.2.102 2.2.100 2.1.0 retired 1.3.0 retired 1.2.1 retired 1.2.0 retired 1.1.0 retired 1.0.0 retired

Tensor library for Gleam/BEAM with a pure Gleam API, zero-copy views, and optional native acceleration

HexDocs

Current section

Files

Jump to
Readme
10 Versions
7 Dependencies
Dependants
Files
Activity
viva_tensor include viva_tensor@nn@transformer_DecoderBlock.hrl
Raw

include/viva_tensor@nn@transformer_DecoderBlock.hrl 

-record(decoder_block, {

    self_attention :: viva_tensor@nn@attention:multi_head_attention(),

    cross_attention :: viva_tensor@nn@attention:multi_head_attention(),

    ffn :: viva_tensor@nn@transformer:feed_forward(),

    norm1 :: viva_tensor@nn@norm:layer_norm(),

    norm2 :: viva_tensor@nn@norm:layer_norm(),

    norm3 :: viva_tensor@nn@norm:layer_norm()

}).