Enum ScaledDotProductAttentionMask

pub enum ScaledDotProductAttentionMask<'a> {
    Array(&'a Array),
    Arrays(&'a [Array]),
    Causal,
}

Mask modes for scaled dot product attention.

Variants

Array(&'a Array)

Array

Arrays(&'a [Array])

Arrays

Causal

Causal

Trait Implementations

impl<'a> Debug for ScaledDotProductAttentionMask<'a>

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

Formats the value using the given formatter.

impl<'a> From<&'a [Array]> for ScaledDotProductAttentionMask<'a>

fn from(masks: &'a [Array]) -> Self

Converts to this type from the input type.

impl<'a> From<&'a Array> for ScaledDotProductAttentionMask<'a>

fn from(mask: &'a Array) -> Self

Converts to this type from the input type.

impl<'a> IntoOption<ScaledDotProductAttentionMask<'a>> for &'a [Array]

fn into_option(self) -> Option<ScaledDotProductAttentionMask<'a>>

Convert into an Option.

impl<'a> IntoOption<ScaledDotProductAttentionMask<'a>> for &'a Array

fn into_option(self) -> Option<ScaledDotProductAttentionMask<'a>>

Convert into an Option.