Expand description
CPU tensor primitives for ruPRIM.
Modulesยง
- binary
- Binary tensor operations (add, sub, mul, div).
- boolean
- cast
- cat
- Concatenation operations for FlexTensor.
- comparison
- Comparison operations returning boolean tensors.
- cumulative
- Cumulative operations along a dimension.
- expand
- Expand operation for broadcasting tensors to larger shapes.
- fill
- flip
- Flip operation for reversing tensor elements along axes.
- gather_
scatter - Gather and scatter operations for indexed tensor access.
- mask
- Mask operations for conditional element replacement.
- quantization
- reduce
- Reduction operations for FlexTensor.
- repeat_
dim - Repeat a tensor along a dimension.
- simd
- SIMD-optimized kernels for tensor operations.
- slice
- Slice operations for FlexTensor.
- sort
- Sort and argsort operations for FlexTensor.
- unary
- Unary tensor operations (exp, log, sqrt, sin, cos, etc.).
- unfold
- Unfold operation for sliding window extraction.