Tensor Library

High-performance operations for N-dimensional arrays.

Operations

All operations support automatic differentiation (autograd).

• matmul(a, b): Matrix multiplication.
• conv2d(input, weight): 2D Convolution.
• relu(x): Activation function.