Struct EdgeFeaturedAttention 

pub struct EdgeFeaturedAttention { /* private fields */ }

Edge-featured attention (GATv2-style)

Implementations

impl EdgeFeaturedAttention

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

impl EdgeFeaturedAttention

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

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

impl EdgeFeaturedAttention

pub fn new(config: EdgeFeaturedConfig) -> Self

Create a new edge-featured attention instance

Arguments

• config - Edge-featured attention configuration

pub fn simple(node_dim: u32, edge_dim: u32, num_heads: u32) -> Self

Create with simple parameters

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

Compute attention without edge features (standard attention)

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

Compute attention with edge features

Arguments

• query - Query vector
• keys - Array of key vectors
• values - Array of value vectors
• edge_features - Array of edge feature vectors (same length as keys)

pub fn node_dim(&self) -> u32

Get the node dimension

pub fn edge_dim(&self) -> u32

Get the edge dimension

pub fn num_heads(&self) -> u32

Get the number of heads

Trait Implementations

impl FromNapiMutRef for EdgeFeaturedAttention

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

Safety

impl FromNapiRef for EdgeFeaturedAttention

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

Safety

impl FromNapiValue for &EdgeFeaturedAttention

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 EdgeFeaturedAttention

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 EdgeFeaturedAttention

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

impl ToNapiValue for EdgeFeaturedAttention

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

Safety

impl TypeName for &EdgeFeaturedAttention

fn type_name() -> &'static str

fn value_type() -> ValueType

impl TypeName for &mut EdgeFeaturedAttention

fn type_name() -> &'static str

fn value_type() -> ValueType

impl TypeName for EdgeFeaturedAttention

fn type_name() -> &'static str

fn value_type() -> ValueType

impl ValidateNapiValue for &EdgeFeaturedAttention

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

Safety

impl ValidateNapiValue for &mut EdgeFeaturedAttention

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

Safety