Struct AlibiInput

pub struct AlibiInput<'a> {
    pub attention_scores: &'a Array,
    pub offset: i32,
    pub mask: Option<&'a Array>,
}

Input for the Alibi module.

Fields

attention_scores: &'a Array

The attention scores.

offset: i32

Offset

mask: Option<&'a Array>

Mask

Implementations

impl<'a> AlibiInput<'a>

pub fn new(attention_scores: impl Into<&'a Array>) -> Self

Creates a new instance of AlibiInput.

impl AlibiInput<'_>

pub const DEFAULT_OFFSET: i32 = 0i32

Default value for offset field.

Trait Implementations

impl<'a> Buildable for AlibiInput<'a>

type Builder = AlibiInputBuilder<'a>

The builder type for this buildable type

impl<'a> Builder<AlibiInput<'a>> for AlibiInputBuilder<'a>

type Error = Infallible

Error with building

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

Build the type

impl<'a> Clone for AlibiInput<'a>

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

Returns a copy of the value.

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

Performs copy-assignment from source.

impl<'a> Debug for AlibiInput<'a>

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

Formats the value using the given formatter.

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

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

Converts to this type from the input type.

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

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

Converts to this type from the input type.

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

fn from((attention_scores, offset): (&'a Array, i32)) -> Self

Converts to this type from the input type.

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

fn from((attention_scores, offset, mask): (&'a Array, i32, &'a Array)) -> Self

Converts to this type from the input type.

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

fn from( (attention_scores, offset, mask): (&'a Array, i32, Option<&'a Array>), ) -> Self

Converts to this type from the input type.