Struct MultiHeadAttentionInput

pub struct MultiHeadAttentionInput<'a> {
    pub queries: &'a Array,
    pub keys: &'a Array,
    pub values: &'a Array,
    pub mask: Option<&'a Array>,
}

Input to the MultiHeadAttention module

Fields

queries: &'a Array

Queries

keys: &'a Array

Keys

values: &'a Array

Values

mask: Option<&'a Array>

Mask

Implementations

impl<'a> MultiHeadAttentionInput<'a>

pub fn new( queries: impl Into<&'a Array>, keys: impl Into<&'a Array>, values: impl Into<&'a Array>, ) -> Self

Creates a new instance of MultiHeadAttentionInput.

Trait Implementations

impl<'a> Buildable for MultiHeadAttentionInput<'a>

type Builder = MultiHeadAttentionInputBuilder<'a>

The builder type for this buildable type

impl<'a> Builder<MultiHeadAttentionInput<'a>> for MultiHeadAttentionInputBuilder<'a>

type Error = Infallible

Error with building

fn build(self) -> Result<MultiHeadAttentionInput<'a>, Self::Error>

Build the type

impl<'a> Clone for MultiHeadAttentionInput<'a>

fn clone(&self) -> MultiHeadAttentionInput<'a>

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl<'a> Debug for MultiHeadAttentionInput<'a>

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

Formats the value using the given formatter.

impl<'a> From<(&'a Array, &'a Array, &'a Array)> for MultiHeadAttentionInput<'a>

fn from((queries, keys, values): (&'a Array, &'a Array, &'a Array)) -> Self

Converts to this type from the input type.

impl<'a> From<(&'a Array, &'a Array, &'a Array, &'a Array)> for MultiHeadAttentionInput<'a>

fn from( (queries, keys, values, mask): (&'a Array, &'a Array, &'a Array, &'a Array), ) -> Self

Converts to this type from the input type.

impl<'a> From<(&'a Array, &'a Array, &'a Array, Option<&'a Array>)> for MultiHeadAttentionInput<'a>

fn from( (queries, keys, values, mask): (&'a Array, &'a Array, &'a Array, Option<&'a Array>), ) -> Self

Converts to this type from the input type.