pub struct GraphAttention { /* private fields */ }Expand description
Graph Attention kernel.
Implements Graph Attention Networks (GAT) with multi-head attention. Learns to weight neighbor contributions based on their relevance to each node.
Implementations§
Source§impl GraphAttention
impl GraphAttention
Sourcepub fn compute(
graph: &CsrGraph,
node_features: &[Vec<f64>],
weights: &GATWeights,
config: &GraphAttentionConfig,
) -> GATResult
pub fn compute( graph: &CsrGraph, node_features: &[Vec<f64>], weights: &GATWeights, config: &GraphAttentionConfig, ) -> GATResult
Compute graph attention layer.
Trait Implementations§
Source§impl Clone for GraphAttention
impl Clone for GraphAttention
Source§fn clone(&self) -> GraphAttention
fn clone(&self) -> GraphAttention
Returns a duplicate of the value. Read more
1.0.0 · Source§fn clone_from(&mut self, source: &Self)
fn clone_from(&mut self, source: &Self)
Performs copy-assignment from
source. Read moreSource§impl Debug for GraphAttention
impl Debug for GraphAttention
Source§impl Default for GraphAttention
impl Default for GraphAttention
Source§impl GpuKernel for GraphAttention
impl GpuKernel for GraphAttention
Source§fn metadata(&self) -> &KernelMetadata
fn metadata(&self) -> &KernelMetadata
Returns the kernel metadata.
Source§fn requires_gpu_native(&self) -> bool
fn requires_gpu_native(&self) -> bool
Returns true if this kernel requires GPU-native execution.
Source§fn health_check(&self) -> HealthStatus
fn health_check(&self) -> HealthStatus
Perform a health check on this kernel. Read more
Source§fn refresh_config(&mut self, _config: &KernelConfig) -> Result<(), KernelError>
fn refresh_config(&mut self, _config: &KernelConfig) -> Result<(), KernelError>
Refresh kernel configuration at runtime. Read more
Auto Trait Implementations§
impl Freeze for GraphAttention
impl RefUnwindSafe for GraphAttention
impl Send for GraphAttention
impl Sync for GraphAttention
impl Unpin for GraphAttention
impl UnwindSafe for GraphAttention
Blanket Implementations§
Source§impl<T> ArchivePointee for T
impl<T> ArchivePointee for T
Source§type ArchivedMetadata = ()
type ArchivedMetadata = ()
The archived version of the pointer metadata for this type.
Source§fn pointer_metadata(
_: &<T as ArchivePointee>::ArchivedMetadata,
) -> <T as Pointee>::Metadata
fn pointer_metadata( _: &<T as ArchivePointee>::ArchivedMetadata, ) -> <T as Pointee>::Metadata
Converts some archived metadata to the pointer metadata for itself.
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
Source§impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
Source§impl<F, W, T, D> Deserialize<With<T, W>, D> for F
impl<F, W, T, D> Deserialize<With<T, W>, D> for F
Source§impl<T> Instrument for T
impl<T> Instrument for T
Source§fn instrument(self, span: Span) -> Instrumented<Self>
fn instrument(self, span: Span) -> Instrumented<Self>
Source§fn in_current_span(self) -> Instrumented<Self>
fn in_current_span(self) -> Instrumented<Self>
Source§impl<T> LayoutRaw for T
impl<T> LayoutRaw for T
Source§fn layout_raw(_: <T as Pointee>::Metadata) -> Result<Layout, LayoutError>
fn layout_raw(_: <T as Pointee>::Metadata) -> Result<Layout, LayoutError>
Gets the layout of the type.