Module ops

Module ops 

Source
Expand description

Tensor operations

Modulesยง

activation
Activation functions
arithmetic
Element-wise arithmetic operations
conv
Optimized convolution operations
matmul
Matrix multiplication and linear algebra operations
reduction
Reduction operations (sum, mean, max, min, etc.)
simd
SIMD-optimized operations for maximum performance