Struct burn_core::nn::transformer::TransformerDecoderInput

pub struct TransformerDecoderInput<B: Backend> { /* private fields */ }

Transformer Decoder forward pass input argument.

Implementations

impl<B: Backend> TransformerDecoderInput<B>

pub fn new(target: Tensor<B, 3>, memory: Tensor<B, 3>) -> Self

Create a transformer decoder input argument.

pub fn memory_mask_pad(self, mask_pad: Tensor<B, 2, Bool>) -> Self

Register the memory padding mask.

pub fn memory_mask_attn(self, mask_attn: Tensor<B, 3, Bool>) -> Self

Register the memory attention mask.

pub fn target_mask_pad(self, mask_pad: Tensor<B, 2, Bool>) -> Self

Register the target padding mask.

pub fn target_mask_attn(self, mask_attn: Tensor<B, 3, Bool>) -> Self

Register the target attention mask.

Trait Implementations

impl<B: Debug + Backend> Debug for TransformerDecoderInput<B>

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.