Skip to main content

Module tensor

Module tensor 

Source
Expand description

Tensor operations for GNN computations.

Provides efficient tensor operations including:

  • Matrix multiplication
  • Element-wise operations
  • Activation functions
  • Weight initialization
  • Normalization

Structs§

Tensor
Basic tensor operations for GNN computations

Functions§

hadamard_product
Element-wise (Hadamard) product
he_init
He initialization for ReLU networks
vector_add
Element-wise vector addition
vector_scale
Scalar multiplication of a vector
xavier_init
Xavier/Glorot initialization for neural network weights