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