yscv-tensor 0.1.8

SIMD-accelerated tensor library with f32/f16/bf16 support and 80+ operations
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50

use thiserror::Error;

/// Errors returned by tensor construction and math operations.
#[derive(Debug, Clone, PartialEq, Eq, Error)]
pub enum TensorError {
    #[error("tensor size overflow for shape {shape:?}")]
    SizeOverflow { shape: Vec<usize> },
    #[error(
        "tensor shape {shape:?} expects {} elements, got {data_len}",
        shape_element_count(shape).unwrap_or(0)
    )]
    SizeMismatch { shape: Vec<usize>, data_len: usize },
    #[error("shape mismatch: left={left:?}, right={right:?}")]
    ShapeMismatch { left: Vec<usize>, right: Vec<usize> },
    #[error("broadcast mismatch: left={left:?}, right={right:?}")]
    BroadcastIncompatible { left: Vec<usize>, right: Vec<usize> },
    #[error("cannot reshape from {from:?} to {to:?} due to size mismatch")]
    ReshapeSizeMismatch { from: Vec<usize>, to: Vec<usize> },
    #[error("axis {axis} is out of range for rank {rank}")]
    InvalidAxis { axis: usize, rank: usize },
    #[error("index rank mismatch: expected {expected}, got {got}")]
    InvalidIndexRank { expected: usize, got: usize },
    #[error("index out of bounds at axis {axis}: index={index}, dim={dim}")]
    IndexOutOfBounds {
        axis: usize,
        index: usize,
        dim: usize,
    },
    #[error("dtype mismatch: expected {expected:?}, got {got:?}")]
    DTypeMismatch { expected: DType, got: DType },
    #[error("unsupported operation: {msg}")]
    UnsupportedOperation { msg: String },
}

/// Element data type for tensor storage.
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
pub enum DType {
    /// 32-bit IEEE 754 float.
    F32,
    /// 16-bit IEEE 754 half-precision float (stored as u16 bit pattern).
    F16,
    /// 16-bit Brain float (stored as u16 bit pattern).
    BF16,
}

fn shape_element_count(shape: &[usize]) -> Option<usize> {
    shape
        .iter()
        .try_fold(1usize, |acc, dim| acc.checked_mul(*dim))
}