Struct AttentionConfig 

pub struct AttentionConfig {
    pub dim: u32,
    pub num_heads: Option<u32>,
    pub dropout: Option<f64>,
    pub scale: Option<f64>,
    pub causal: Option<bool>,
}

Attention configuration object

Fields

dim: u32

num_heads: Option<u32>

dropout: Option<f64>

scale: Option<f64>

causal: Option<bool>

Trait Implementations

impl FromNapiValue for AttentionConfig

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

Safety

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

impl ToNapiValue for AttentionConfig

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

Safety

impl TypeName for AttentionConfig

fn type_name() -> &'static str

fn value_type() -> ValueType

impl ValidateNapiValue for AttentionConfig

unsafe fn validate( env: *mut napi_env__, napi_val: *mut napi_value__, ) -> Result<*mut napi_value__, Error>

Safety