Struct burn_core::nn::transformer::TransformerEncoderInput

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

Transformer Encoder forward pass input argument.

Implementations

impl<B: Backend> TransformerEncoderInput<B>

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

Create a transformer encoder input argument.

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

Register the padding mask.

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

Register the attention mask.

Trait Implementations

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

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

Formats the value using the given formatter.