Struct DotProductAttention 

pub struct DotProductAttention { /* private fields */ }

Scaled dot-product attention

Implementations

impl DotProductAttention

pub fn instance_of<V: NapiRaw>(env: Env, value: V) -> Result<bool>

impl DotProductAttention

pub fn into_reference( val: DotProductAttention, env: Env, ) -> Result<Reference<DotProductAttention>>

pub fn into_instance( self, env: Env, ) -> Result<ClassInstance<DotProductAttention>>

impl DotProductAttention

pub fn new(dim: u32) -> Result<Self>

Create a new scaled dot-product attention instance

Arguments

• dim - Embedding dimension

pub fn compute( &self, query: Float32Array, keys: Vec<Float32Array>, values: Vec<Float32Array>, ) -> Result<Float32Array>

Compute attention output

Arguments

• query - Query vector
• keys - Array of key vectors
• values - Array of value vectors

pub fn compute_with_mask( &self, query: Float32Array, keys: Vec<Float32Array>, values: Vec<Float32Array>, mask: Vec<bool>, ) -> Result<Float32Array>

Compute attention with mask

Arguments

• query - Query vector
• keys - Array of key vectors
• values - Array of value vectors
• mask - Boolean mask array (true = attend, false = mask)

pub fn dim(&self) -> u32

Get the dimension

Trait Implementations

impl FromNapiMutRef for DotProductAttention

unsafe fn from_napi_mut_ref( env: napi_env, napi_val: napi_value, ) -> Result<&'static mut Self>

Safety

impl FromNapiRef for DotProductAttention

unsafe fn from_napi_ref( env: napi_env, napi_val: napi_value, ) -> Result<&'static Self>

Safety

impl FromNapiValue for &DotProductAttention

unsafe fn from_napi_value(env: napi_env, napi_val: napi_value) -> Result<Self>

Safety

fn from_unknown(value: JsUnknown) -> Result<Self, Error>

impl FromNapiValue for &mut DotProductAttention

unsafe fn from_napi_value(env: napi_env, napi_val: napi_value) -> Result<Self>

Safety

fn from_unknown(value: JsUnknown) -> Result<Self, Error>

impl ObjectFinalize for DotProductAttention

fn finalize(self, env: Env) -> Result<(), Error>

impl ToNapiValue for DotProductAttention

unsafe fn to_napi_value( env: napi_env, val: DotProductAttention, ) -> Result<napi_value>

Safety

impl TypeName for &DotProductAttention

fn type_name() -> &'static str

fn value_type() -> ValueType

impl TypeName for &mut DotProductAttention

fn type_name() -> &'static str

fn value_type() -> ValueType

impl TypeName for DotProductAttention

fn type_name() -> &'static str

fn value_type() -> ValueType

impl ValidateNapiValue for &DotProductAttention

unsafe fn validate(env: napi_env, napi_val: napi_value) -> Result<napi_value>

Safety

impl ValidateNapiValue for &mut DotProductAttention

unsafe fn validate(env: napi_env, napi_val: napi_value) -> Result<napi_value>

Safety