Struct TensorData 

pub struct TensorData {
    pub shape: Vec<usize>,
    pub dtype: String,
    pub data: Vec<u8>,
    pub requires_grad: bool,
}

Tensor data with shape and values.

Fields

shape: Vec<usize>

Tensor shape (dimensions)

dtype: String

Data type identifier

data: Vec<u8>

Flattened tensor values (stored as bytes)

requires_grad: bool

Whether this tensor requires gradient

Implementations

impl TensorData

pub fn from_f32(shape: Vec<usize>, data: &[f32]) -> Self

Create new tensor data from f32 values.

pub fn from_f64(shape: Vec<usize>, data: &[f64]) -> Self

Create new tensor data from f64 values.

pub fn as_f32(&self) -> Result<Vec<f32>>

Get tensor as f32 slice.

pub fn as_f64(&self) -> Result<Vec<f64>>

Get tensor as f64 slice.

pub fn num_elements(&self) -> usize

Get number of elements.

Trait Implementations

impl Clone for TensorData

fn clone(&self) -> TensorData

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for TensorData

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for TensorData

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for TensorData

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.