Struct Tensor 

pub struct Tensor {
    pub name: String,
    pub dtype: DType,
    pub shape: Vec<u32>,
    pub data: Vec<u8>,
}

A tensor with shape and data.

Fields

name: String

Tensor name

dtype: DType

Data type

shape: Vec<u32>

Shape dimensions

data: Vec<u8>

Raw data bytes

Implementations

impl Tensor

pub fn new( name: impl Into<String>, dtype: DType, shape: Vec<u32>, data: Vec<u8>, ) -> Self

Create a new tensor.

pub fn numel(&self) -> usize

Get number of elements.

pub fn expected_size(&self) -> usize

Get expected data size in bytes.

pub fn is_valid(&self) -> bool

Validate tensor data size.

pub fn to_f32_vec(&self) -> Option<Vec<f32>>

Get data as f32 vector (if dtype is F32).

pub fn from_f32(name: impl Into<String>, shape: Vec<u32>, data: &[f32]) -> Self

Create f32 tensor from slice.

Trait Implementations

impl Clone for Tensor

fn clone(&self) -> Tensor

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for Tensor

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.