Index

A

attention_map (encoder.Encoder property) (multiHeadAttention.MultiHeadAttention property)

D

decoder module

Decoder (class in decoder)

E

encoder module

Encoder (class in encoder)

F

forward() (decoder.Decoder method) (encoder.Encoder method) (loss.OZELoss method) (multiHeadAttention.MultiHeadAttention method) (multiHeadAttention.MultiHeadAttentionChunk method) (multiHeadAttention.MultiHeadAttentionWindow method) (positionwiseFeedForward.PositionwiseFeedForward method) (transformer.Transformer method)

G

generate_local_map_mask() (in module utils)

generate_original_PE() (in module utils) generate_regular_PE() (in module utils)

L

loss module

M

module decoder encoder loss multiHeadAttention positionwiseFeedForward transformer utils

multiHeadAttention module MultiHeadAttention (class in multiHeadAttention) MultiHeadAttentionChunk (class in multiHeadAttention) MultiHeadAttentionWindow (class in multiHeadAttention)

O

OZELoss (class in loss)

P

positionwiseFeedForward module

PositionwiseFeedForward (class in positionwiseFeedForward)

T

transformer module

Transformer (class in transformer)

U

utils module